TPU Kernel Engineer
Designs and optimizes TPU kernels to address performance issues in ML research, training, and inference systems. Provides feedback on model impacts and solves large-scale systems problems, requiring deep accelerator expertise.
Lead a team developing large-scale RL and ML models for autonomous vehicle behavior planning and driving decisions. Requires RL/ML expertise, production ML pipeline experience, and 3+ years in leadership.
Designs and optimizes TPU kernels to address performance issues in ML research, training, and inference systems. Provides feedback on model impacts and solves large-scale systems problems, requiring deep accelerator expertise.
Deploys state-of-the-art ML models and fine-tunes LLMs in production to enhance platform integrity and safety against adversarial threats. Requires Master's/PhD, deep learning expertise, PyTorch/TensorFlow proficiency.
Software Engineer enabling production AI workloads on new hardware platforms through porting, benchmarking, stress testing, and performance optimization. Requires 5+ years in ML systems, distributed training, PyTorch, and RDMA/NCCL expertise.
Develop and deploy retrieval and search algorithms for OpenAI's API and ChatGPT, collaborating with research teams on production systems for millions of users. Requires experience with ML systems, vector databases, and large-scale search.
Distributed Training Engineer optimizes training throughput for video foundation models, enabling researcher experiments with efficient architectures. Requires strong Python skills, distributed systems expertise, and performance profiling.