optimization Jobs

Prime Intellect is building the open superintelligence stack, providing infrastructure for AI labs. Our platform, Lab, unifies compute, environments, evaluations, secure sandboxes, high-performance training, and deployment for post-training at frontier scale. We are developing open-source models trained end-to-end for long-horizon tasks like autonomous research, alongside the full-stack platform our research team utilizes. We are seeking individuals passionate about building at the intersection of frontier research, real infrastructure, and go-to-market for a nascent category.

$150k - $350k

NVIDIA is seeking interns to research scalable differentiable algorithms, including adjoint and automatic differentiation methods. This role will involve connecting simulations with experimental data for calibration and data assimilation using AI/ML techniques. The work will be evaluated on simulation workloads and assessed for integration into NVIDIA software. This internship is ideal for individuals passionate about mathematics, code, hardware, and engineering applications.

$0k - $0k

Join our Conversation Engine team to scale and optimize a cutting-edge Generative AI product that provides instant assistance to enterprise users. You will work at the core of Moveworks' Generative AI product, focusing on scaling and optimizing it for the enterprise solution space. We encourage innovative ideas to solve unsolved problems, collaborating closely with machine learning experts and cross-functional teams to rapidly iterate on new ideas and leverage user behavior data. Your challenge will be to scale and optimize our conversation engine to support hundreds of millions of users across multiple chat channels and use cases, enhancing our end-to-end product infrastructure with engineering quality and robustness.

$176k - $308k

Anthropic

Anthropic is dedicated to building AI systems that are safe, reliable, and steerable, aiming to benefit both users and society. This role is central to advancing Reinforcement Learning (RL), the core technology enabling Claude's reasoning, coding, and autonomous capabilities. The RL Scaling team focuses on how RL performance, stability, and learning efficiency evolve with increasing model size, episode length, and compute power. You will be instrumental in developing and scaling next-generation RL algorithms and architectures, from initial research to large-scale deployment, while deeply understanding and addressing the challenges that arise at scale. This position bridges research and engineering, requiring you to build the systems that accelerate iteration and experimentation, and to tackle complex, unsolved problems in Anthropic's largest RL runs.

$500k - $850k

Joining NVIDIA's DGX Cloud AI Efficiency Team means advancing the performance, efficiency, and resiliency of large-scale AI workloads. You will help AI researchers and platform teams understand end-to-end behavior across GPUs, networking, storage, and software stacks. Your work will shape scalable DGX Cloud systems, turn complex measurements into prioritized engineering decisions, and continuously raise the performance and reliability of AI workloads. Join a technically diverse team of infrastructure experts to unlock more efficient AI at scale.

$224k - $431k

Together AI

As a Systems Research Engineer specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications. You will co-design GPU kernels and model architecture to enhance the performance and efficiency of our AI systems, and contribute to the co-design of efficient GPU architectures and programming models. Your research skills will be vital in staying up-to-date with the latest advancements in GPU programming techniques, ensuring that our AI infrastructure remains at the forefront of innovation.

$200k - $300k

Databricks

As a Senior Applied ML Engineer on the Applied AI team, you will leverage machine learning, scheduling, and optimization algorithms to enhance the efficiency and performance of Databricks' infrastructure. Your responsibilities will encompass the full technology stack, from cluster management to query compilation, tackling complex, high-impact engineering challenges to deliver optimized and cost-effective workloads for customers. This role is crucial for accelerating serverless growth, building end-to-end ML4Sys solutions, shaping applied ML strategy, and driving the deployment of state-of-the-art models.

$166k - $210k

The Conversation Engine team is seeking a Staff Software Engineer to scale and optimize a cutting-edge Generative AI product that provides instant assistance to enterprise users. This role offers a unique opportunity to work at the core of Moveworks' Generative AI product, focusing on its scalability and optimization for the enterprise solution space. You will be encouraged to bring innovative ideas to solve complex problems, collaborating closely with machine learning experts and cross-functional teams. The challenge involves scaling the conversation engine to support hundreds of millions of users across multiple chat channels and use cases, while enhancing the end-to-end product infrastructure with high engineering quality and robustness.

$176k - $308k