Simplify Logo

Full-Time

Staff/Senior Staff Software Engineer

ML Performance Optimization

Confirmed live in the last 24 hours

Zoox

Zoox

1,001-5,000 employees

Develops autonomous electric ride-hailing vehicles

Energy
AI & Machine Learning
Automotive & Transportation

Compensation Overview

$234k - $342kAnnually

+ Sign-on Bonus + Amazon RSUs + Zoox Stock Appreciation Rights

Senior, Expert

San Mateo, CA, USA

Position requires onsite presence in Foster City, CA.

Category
Backend Engineering
Embedded Engineering
Software Engineering
Required Skills
Python
Requirements
  • 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's Nsight or PyTorch's 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.
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 x-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.

Zoox is developing a fully integrated autonomous ride-hailing service with a vehicle designed specifically for self-driving. This vehicle features advanced sensors and AI systems for navigating urban environments and can comfortably accommodate four passengers. Unlike competitors that retrofit existing cars, Zoox's approach allows for optimized safety and efficiency, with a strong commitment to sustainability through fully electric vehicles. The goal is to enhance urban transportation, making it smarter and safer while improving the quality of life in cities.

Company Stage

M&A

Total Funding

$2.3B

Headquarters

Menlo Park, California

Founded

2014

Growth & Insights
Headcount

6 month growth

5%

1 year growth

21%

2 year growth

44%
Simplify Jobs

Simplify's Take

What believers are saying

  • Zoox's association with Amazon provides strong financial backing and access to advanced technologies, enhancing job security and innovation potential.
  • The company's rapid expansion into new markets like Austin and Miami offers employees opportunities for career growth and new project experiences.
  • Zoox's focus on purpose-built autonomous vehicles positions it at the forefront of the self-driving car industry, offering employees a chance to work on cutting-edge technology.

What critics are saying

  • Incidents involving Zoox's self-driving vehicles, such as unexpected braking and crashes, could lead to regulatory scrutiny and operational setbacks.
  • The competitive landscape with established players like Waymo and Cruise may pressure Zoox to accelerate its development timeline, potentially compromising safety and quality.

What makes Zoox unique

  • Zoox's purpose-built autonomous robotaxi is designed specifically for self-driving, unlike retrofitted vehicles used by some competitors.
  • Backed by Amazon, Zoox has significant financial and technological resources to accelerate its development and deployment of autonomous vehicles.
  • Zoox's expansion into multiple cities like Austin, Miami, and Seattle demonstrates its aggressive growth strategy and commitment to broadening its operational footprint.

Benefits

Health Insurance

Maternity & Paternity Leave

Vacation & Paid Time Off

Sick Days

Free Lunch or Snacks

Employee Assitance Program