1 day ago
Foster City, CA, USAMid Level / Senior
H1B Sponsor
Base Salary
$192k - $257k/yr
Responsibilities
- Design, implement, and operate ML performance optimization techniques for training and inference.
- Collaborate with ML researchers, software engineers, data engineers, and hardware engineers.
- Define requirements and align on architectural decisions across teams.
Requirements
- 4+ years of total experience, including 2+ years on large-scale model training or inference platforms.
- Experience with training frameworks like PyTorch and efficient GPU utilization.
- Experience with GPU-accelerated inference using TensorRT or similar frameworks.
- Familiarity with profiling tools like NVIDIA's Nsight or PyTorch's Profiler.
- Proficient in Python or C++.