DeepSpeed Jobs

CoreWeave is seeking a Principal Engineer to lead its Benchmarking & Performance team. This role will be responsible for a global performance data warehouse, ingesting, storing, transforming, and analyzing performance events across all data centers. You will also play a key part in industry-leading performance benchmarking publications, working closely with NVIDIA and the open-source community to demonstrate CoreWeave's performance and reliability leadership in AI.

$206k - $333k

Databricks

Databricks is seeking a Senior Staff Software Engineer to work on the AI Runtime system, which powers the world's largest AI platform. This serverless product enables companies to train their own models using enterprise data, similar to how frontier labs operate. The role involves tackling complex technical challenges to advance AI Runtime's architecture, reliability, and developer experience for both humans and AI agents. You will contribute to shaping the long-term technical direction of AI training infrastructure, anticipating industry trends, and optimizing training efficiency and cost across various model architectures and GPU accelerators.

$229k - $343k

Databricks

Databricks is seeking a Senior Software Engineer for its AI Runtime (AIR) team. AIR is a managed platform designed for large-scale GPU training and fine-tuning of AI models, providing customers with on-demand access to accelerators and a serverless experience that simplifies the complexities of provisioning, scheduling, and orchestrating multi-node jobs. As a Senior Software Engineer on AIR, you will be instrumental in building and scaling the systems that ensure large-scale training is fast, reliable, and effortless. You will lead the architecture and evolution of the managed GPU training stack, focusing on areas such as scheduling, distributed training performance, fault tolerance, and the developer experience for large-scale job operations. Beyond hands-on system development, you will contribute to the technical vision for AIR, mentor engineers, and collaborate with product, research, and platform teams to expand the impact of custom AI training at Databricks.

$160k - $225k

Databricks

Databricks is seeking a Staff Software Engineer for its AI Runtime (AIR) team. AIR is a managed platform designed for large-scale GPU training and fine-tuning of state-of-the-art AI models, providing customers with on-demand access to accelerators and a serverless experience. As part of this role, you will be instrumental in building and scaling the systems that ensure fast, reliable, and effortless large-scale training. You will influence the architecture and development of the managed GPU training stack, focusing on areas like scheduling, distributed training performance, fault tolerance, and the overall developer experience for operating jobs at scale. This position also involves defining the long-term technical vision for AIR, mentoring senior engineers, and collaborating with product, research, and platform teams to expand the impact of custom training at Databricks.

$190k - $265k

Prime Intellect is building the open superintelligence stack, providing infrastructure for AI labs. Their platform, Lab, unifies compute, environments, evaluations, and training/deployment for post-training at scale, including SFT, RL, and agent workflows. They are developing open-source models for long-horizon tasks and the platform their research team uses. The company has raised significant funding and is looking for individuals passionate about frontier research, infrastructure, and go-to-market strategies in a nascent category.

$150k - $350k

Aleph Alpha is seeking a Senior AI Researcher to join their Pre-training team, focusing on advancing the architecture and training of next-generation foundation models. This role involves designing inference-efficient architectures, optimizing large-scale training recipes, and working with thousands of NVIDIA GPUs. The position is ideal for someone excited about the intersection of research and engineering, translating mathematical reasoning into principled training decisions from small-scale experiments to large-scale runs. The researcher will influence model quality, run reliability, and efficiency by combining rigorous experimental design with high-quality production code.

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance AI research, cultivate the next generation of AI builders, and make significant contributions to a knowledge-driven economy. As part of our team, you will work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, contributing to groundbreaking AI solutions that have the potential to reshape industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy. As part of our team, you will have the opportunity to work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, participating in the creation of groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

SambaNova is seeking a talented and driven ML performance engineer to optimize and scale state-of-the-art foundation models on SambaNova's reconfigurable dataflow platform. You will work hands-on with advanced models to push the limits of throughput, latency, and efficiency. This role bridges the gap between deep learning and systems performance, collaborating across compiler, runtime, and hardware layers to deliver world-record performance for large-scale AI inference.

$180k - $255k

Nunchux AI is building infrastructure to make multimodal generative AI faster, cheaper, and easier to build upon. We are seeking a Machine Learning Engineer focused on post-training and evaluation to enhance the efficiency and quality of visual generative models. In this role, you will develop and validate post-training techniques, curate and manage data pipelines for training and evaluation, and build robust evaluation systems to measure model performance. You will also benchmark models against industry standards, ensure quality, and integrate cutting-edge research into production workflows.

$180k - $250k