Staff Software Engineer, ML Performance Optimization
Leads ML performance optimization for training and inference platforms in autonomous driving, collaborating across teams to enhance efficiency using PyTorch, TensorRT, and profiling tools. Requires strong Python/C++ skills, GPU expertise, and 4+ years in large-scale ML platforms; leads engineering team.
242k – 389k/yr
On-site4+ YOEML Engineering
About the role
Responsibilities
Develop and execute a strategic vision for the ML Performance Optimization team to unlock ML innovation in autonomous driving and rider experience.
Lead the design, implementation, and operation of cutting-edge ML Training and inference performance optimization techniques.
Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions.
Enable the engineers in the team to grow their careers by providing technical guidance and mentorship.
Qualifications
Strong experience with training frameworks like PyTorch, leveraging GPUs efficiently for distributed model training.
Experience with GPU-accelerated inference using TensorRT, Ray Serve, or similar frameworks.
Experience using profiling tools like NVIDIA Nsight or PyTorch Profiler for identifying model training and serving bottlenecks.
Proficient in Python and C++.
Experience with model compression techniques to reduce model size and improve performance.
Bonus Qualifications
10+ years of total experience, including 4+ years of working on large-scale model training or inference platforms.
Excellent leadership skills with a demonstrated ability to lead high-performing engineering teams.
Develops and maintains behavioral models for road users (vehicles, pedestrians, cyclists) in autonomous driving perception stack. Requires MS/PhD, 7+ years experience, deep learning expertise, Python fluency, and production ML pipelines.
Leads ML initiatives to enhance validation processes using large-scale fleet and synthetic data for autonomous vehicle safety. Requires PhD or 5+ years ML experience, Python/PyTorch expertise, and AV domain knowledge.
242k – 333k/yr
Hybrid5+ YOEML Engineering
Staff Software Engineer
TrabaNew York, NY +1
Build and scale Traba's applied AI platform for industrial supply chain staffing and operations. Architect core backend systems integrating frontier models and agents for automation, matching, and vetting while owning performance and infrastructure.
240k – 300k/yr
Hybrid7+ YOEML Engineering
Staff Software Engineer
TrabaNew York, NY +1
Build and lead Traba's agentic AI platform as a founding member of the Agents team. Architect orchestration, evals, model strategy, and integrations for autonomous agents in industrial supply chain workflows. Requires 7+ years engineering experience including production LLM/agent systems, with customer immersion and 0-to-1 build experience.
240k – 300k/yr
Hybrid7+ YOEML Engineering
Senior Staff Machine Learning Engineer, Trust
AirbnbUnited States
Senior technical leader building, productionizing and operating large-scale ML models and Agentic AI systems to fight fraud, ensure safety and build trust across Airbnb's platform. Requires 12+ years applied ML experience and deep expertise in LLMs/GenAI.