ML systems Jobs

An applied research team within NVIDIA's Networking Systems & Software Architecture group is tackling complex AI infrastructure challenges. The team develops systems-level software to facilitate high-speed data transfer between GPUs, nodes, and storage, crucial for modern AI demands. This involves optimizing low-level transport, co-designing hardware and software, and creating communication frameworks that integrate with production AI stacks. The team's scope also extends to emerging areas like quantum computing interconnects. The Senior Architect will lead modules and projects from initial research scoping through to production code delivery, acting as a recognized expert to guide technical decisions, incorporate external ideas, and prototype novel solutions.

$224k - $431k

Anthropic

Anthropic is seeking individuals for a paid, fixed-term research residency program, lasting between six weeks and six months. This program is designed for experienced professionals, such as professors in quantitative disciplines, technical founders, or staff-level engineers, who are looking to transition into full-time work focused on ensuring AI systems are safe and beneficial for humanity. While a background in machine learning is not strictly required, candidates should be motivated by AI safety and excited about empirical AI research. The residency offers a scoped project with direct mentorship from Anthropic researchers, access to Claude models and GPU compute, and connections to the AI safety and security research community.

$350k - $500k

Inworld is a research lab and inference provider focused on real-time AI for consumer-facing applications. We build first-party speech models, serve LLMs, and run the inference behind modular APIs designed for high-volume, real-time workloads. Our models and infrastructure support consumer applications across companions, healthcare, fitness, education, media, and more, serving over 10 trillion LLM tokens per month. We are looking for strong individuals from varied backgrounds who learn fast, thrive in ambiguity, and can demonstrate their experience in building, breaking, and understanding complex systems.

$250k - $300k

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 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