Member of Technical Staff, ML Compiler and Systems (Bay Area, Remote)
via Ashby
About this role
What You'll Do
- Lead the evolution of our high-performance robotics simulation platform
- Design and implement the compute infrastructure and data flow mechanisms to optimize performance for physics simulation and foundation model training
- Lead development of our compiler stack, focusing on JIT compilation, LLVM IR, and GPU codegen to minimize compile time and maximize runtime performance
- Collaborate with the team to improve the compiler's support for differentiable programming, crucial for training neural networks within simulations
- Stay current on state-of-the-art ML compilers—such as those in torch, Triton, and JAX—and decide which techniques and approaches are best suited for our application…
What we'd score you on
reqspace match rubricFive dimensions, recruiter-grade. Upload your resume and we'll generate a written explanation of where you fit and where the gaps are.
1
Skills match
For this role: jax
2
Level fit
We check your title trajectory against the seniority signal of the role.
3
Domain experience
Your work in the role's domain matters more than your years total. We weight recent and direct experience.
4
Recency
A skill you used last quarter weighs more than one from five years ago. We grade on recency, not lifetime.
5
Location fit
This role is based in a specific location. We weight your proximity and willingness to relocate.
Score yourself on this role.
Free · no card · written explanation included
Skills in this role
Pulled from the job description. These are the keywords we'll weight when scoring your fit.
jax
More at Genesis-ai
- View →Business Development Manager
- View →Member of Technical Staff, Foundation Models (Bay Area)
- View →Member of Technical Staff, Robot Learning (Bay Area)
- View →Member of Technical Staff, Mechanical Engineer (Bay Area)
- View →Member of Technical Staff, Training (Bay Area, Remote)
- View →Robot Data Operations Analyst