Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo.
Research Engineer, Infrastructure, Kernels
📅 Aug 4
CUDACuteTritonGPU Programming
📜 Description
- Design and implement custom ML kernels (e.g., CUDA, CuTe, Triton) for core LLM operations optimized for modern GPU architectures.
- Collaborate with research teams to align kernel-level optimizations with model architecture and algorithmic goals.
- Develop and maintain a library of reusable kernels and performance benchmarks for internal model training.
- Contribute to infrastructure stability and scalability, ensuring reproducibility and high utilization of compute resources.
- Document and share insights through internal talks, technical papers, or open-source contributions.
🛠️ Requirements
- Bachelor's degree or equivalent experience in computer science, electrical engineering, statistics, machine learning, or similar.
- Strong engineering skills with the ability to contribute performant, maintainable code.
- Understanding of deep learning frameworks (e.g., PyTorch, JAX) and their underlying system architectures.
- Proficiency in CUDA, CuTe, Triton, or other GPU programming frameworks.
- Demonstrated ability to analyze, profile, and optimize compute-intensive workloads.
✨ Benefits
- Dental
- Vision benefits
- Unlimited PTO
- Paid parental leave
- Relocation support as needed
Similar jobs
Search more Research Engineer jobs
Full-Stack Engineer, Human Agent Tooling
Chime👥 10,000+ employees🏢 Financial Services
As a Full-Stack Engineer on the Human Agent Tooling team, you'll develop tools that empower support agents to resolve member issues efficiently and effectively.
Full Stack Software Engineer
SpaceX👥 10,000+ employees🏢 Aviation And Aerospace Component Manufacturing🤝 B2B
As a Full Stack Software Engineer, you will develop mission-critical applications, improve operational efficiency, and collaborate on innovative software solutions at SpaceX.
Corporate AI Engineer
Tanium👥 10,000+ employees🏢 IT Services And IT Consulting🤝 B2B
The Corporate AI Engineer is responsible for building, integrating, and operationalizing AI-enabled solutions that drive measurable business value across the enterprise.
Infrastructure engineer (UK)
Writer👥 501 - 1000 employees🏢 Computer Software
As an Infrastructure Engineer, you'll ensure the availability and reliability of WRITER's platform, tackling complex challenges and driving automation to support enterprise AI workflows.