🕒 2 days ago
Lead the design, training, evaluation, and deployment of large-scale machine learning models, owning training systems and technical decisions from model architecture through production.
Member of Technical Staff, ML Performance
📅 Mar 11
Distributed TrainingInference OptimizationPytorchTensorflow
📜 Description
- Optimize models for real-time use at large scale.
- Design distributed training strategies to reduce training time and resource consumption on large GPU clusters.
- Partner with ML researchers and engineers to develop performant model architectures from conception.
- Build tools to identify performance bottlenecks and stability issues in training and serving.
- Develop frameworks and system designs that improve performance across model development and inference infrastructure.
- Make technical decisions with significant autonomy and use the latest-generation GPUs.
🛠️ Requirements
- 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.
Full job description
Who we are
Odyssey is an AI lab pioneering general world models: causal, multimodal systems that learn to predict and interact with the world over long horizons. This foundational technology promises to revolutionize robotics, science, healthcare, education, gaming, defense, and beyond.
Odyssey’s founders previously pioneered the most complex application of physical AI: self-driving cars. They’ve now brought together a world-class research team from DeepMind, Tesla, Waymo, Meta, Apple, and Wayve, who have made significant contributions to language models (DeepMind Gemini), video models (DeepMind Veo), world models (Wayve GAIA), and autonomous systems (Tesla FSD).
Odyssey has raised significant venture capital from GV, Amazon, AMD, EQT, NVIDIA, Natural Capital, In-Q-Tel, Elad Gil, Jeff Dean, Guillermo Rauch, Garry Tan, Kyle Vogt, and researchers from OpenAI, DeepMind, MSL, Recursive, and Thinking Machines.
What we're looking for
We're seeking those who are obsessed with gaining every last drop of performance from complex systems. We're building inference infrastructure to scale to hundreds of thousands of users within a year, while also working with massive, ever-growing datasets and models in training. Your focus will be ensuring our models deliver exceptional speed, reliability, and scalability in both the training and inference phases, optimizing efficiency to minimize TFLOPS per user and training compute cost.
What you'll do
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.
Who you are
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.
Our staff is expected to be in office at least three times a week. A lot of our progress comes from working through problems together, and we want you in the room for that.
Similar jobs
Search more Machine Learning Engineer jobs
🕒 4 days ago
Own the end-to-end machine learning behind customer-facing products, from training and evaluating models on real-world construction documents to deploying, monitoring, and improving them in production.
Senior Software Engineer, Machine Learning Platform
🕒 10 days ago
Airwallex👥 10,000+ employees🏢 Financial Services🤝 B2B
🕒 10 days ago
Build and scale the machine learning platform behind risk decisioning, developing reliable model and data infrastructure that helps teams deploy, monitor, and improve models protecting transactions across the platform.
🕒 11 days ago
Own the end-to-end development of in-cabin vision models, from data curation and training to evaluation and deployment, supporting passenger understanding, rider safety, and reliable robotaxi operations.
🕒 12 days ago
Own the software and infrastructure that take machine learning models from experiments to large-scale training and production inference, enabling research scientists to build, evaluate, and serve models reliably.