Startups · AI

Member of Technical Staff, ML Performance

Odyssey · Palo Alto · On-site

← All jobs
About Odyssey

Odyssey builds, secures and integrates capabilities and technologies to meet tomorrow’s mission requirements. We enhance operational readiness through comprehensive program support. Backed by a16z, General Catalyst and GV.

About the role

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.

What they're looking for

  • 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
More about this role

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.

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

Read the full posting on Odyssey's site ↗

Engineering

Build your edge while you search

Free tools for founders and investors, plus VC Unfiltered, our take on startups, venture and the people who build them.