Senior / Staff ML Onboard Optimization Engineer
New
Remote US & CanadaFull-TimeSenior
Salary not disclosed
Apply NowOpens the employer's application page
Job Details
- Experience
- Minimum of 6 years of industry experience
- Required Skills
- PythonEmbedded SystemsMachine LearningPyTorchC++Rust
Requirements
- MS/PhD or Bachelors degree in Computer Science, Robotics or similar technical field.
- Minimum 6 years of industry experience.
- Solid coding proficiency in Python, C++ or Rust.
- Experience in deep learning frameworks such as PyTorch.
- Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.
- Experience with Nvidia embedded platforms such as Nvidia Jetson or Thor.
Responsibilities
- Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach.
- Expand the model deployment pipeline to new GPUs and embedded systems for the next generation of our onboard compute system.
- Use frameworks such as TensorRT and modelopt to optimize the models running on the truck.
- Create and benchmark new CUDA kernels for inference.
- Comprehensively profile model runtime and memory to pinpoint performance bottlenecks.
View Full Description & ApplyYou'll be redirected to the employer's site
