← All roles

Member of Technical Staff, ML Performance

Odyssey· Engineering
Apply on company site
Hybrid (Palo Alto), Hybrid (Zurich), Hybrid (London)Full-timePosted 5mo ago

About the role

Senior software engineer focused on maximizing ML performance for real-time inference and large-scale training, optimizing speed, reliability, and scalability across training and inference infrastructures.

Responsibilities

  • Optimize models that will be used in real-time by hundreds of thousands of users.
  • Design and implement distributed training strategies to reduce training time and resource consumption on large GPU clusters.
  • Partner with our elite team of ML researchers and engineers to ensure model architectures are highly performant from conception.
  • Develop sophisticated tools to identify performance bottlenecks and stability issues in both training and serving environments.
  • Pioneer innovative approaches, frameworks, and system designs that enhance performance metrics across our model development and inference infrastructure.
  • Have significant autonomy in technical decisions.
  • Use the latest-generation GPUs.

Qualifications

  • 8+ years of software engineering experience, with significant work in ML performance.
  • Deep insight into modern machine learning architectures with a natural instinct for performance optimization, particularly distributed training and inference.
  • Track record of owning projects end to end.
  • Problem-solving mindset with the ability to acquire new skills as needed.
  • Proficiency with PyTorch (or TF/JAX) and Triton as well as NVIDIA GPU ecosystems and optimization stacks.
  • Highly metric-based.

Skills

Required:ML performance optimizationDistributed trainingPyTorchTritonNVIDIA GPU ecosystems
Last updated Oct 8, 2026. Interested? Apply on the company site.

About Odyssey

Odyssey is an AI lab focused on general-purpose world models: causal, multimodal systems that learn to predict and interact with the world over long horizons

Odyssey is an AI lab pioneering general-purpose world models—a new form of multimodal intelligence.

World ModelOdyssey-2ResearchWritingCareersContactOur ModelsGet Started
Company profileWebsite

More open roles at Odyssey