ML Compiler
About The Position
We're building a helpful robot for every home. We're a small team of engineers, designers, and operators based in San Francisco. Our team comes from Tesla, Cruise, OpenAI, Google, Pixar, and many other great companies. In the past we've shipped to hundreds of millions of users and know what it takes to build amazing products and experiences. Our team is deliberately lean to promote rapid decision making and do away with bureaucracy and hierarchy. Everyone is an IC and is empowered with massive scope, radical ownership, and direct responsibility. We work across the stack with a culture built for rapid iteration and fast execution. As an expert in building ML compilers for edge devices (custom silicon or otherwise), you will play an integral role in a robust deployment stack to run large neural networks with minimal latency on our robots.
Requirements
- Strong coding ability, with exceptional experience with C++ and/or Python.
- Experience with modern compiler infrastructure (MLIR/LLVM, XLA, TVM, Glow, etc).
- Experience with running models on heterogeneous compute (edge devices preferred).
- Experience with writing kernels (CUDA/OpenCL).
Nice To Haves
- Experience with quantization is a plus, but not required.
Responsibilities
- Design, build, and maintain core compiler infrastructure for our target hardware.
- Collaborate with other functions (e.g. ML, Systems Software, etc).
- Independently triage and solve complex numerical issues (disparities between training and inference) while pushing the performance envelope.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Education Level
No Education Listed
