about 1 month ago
Remote, United StatesSenior / Staff+
H1B Sponsor
Base Salary
$180k - $200k/yr
Responsibilities
- Own problems end-to-end from exploratory R&D through production-hardened, maintainable systems.
- Develop and extend GPU-accelerated code in PyTorch and CUDA C++.
- Work across a broad modeling landscape including RL, graph neural networks, and optimization.
- Formulate objectives, model constraints, and debug numerical behavior in the stack.
- Contribute to technical direction and research strategy alongside senior teammates.
Requirements
- 4+ years of industry experience in ML, optimization, or a related field.
- Strong fundamentals in machine learning and optimization.
- Production PyTorch experience.
- Demonstrated ability to work across research and production codebases.
- Comfort operating with high autonomy in ambiguous problem spaces.
- Strong communication and collaboration skills.
Benefits
- Interesting and challenging work.
- Competitive salary and equity benefits.
- Health, dental, and vision insurance.
- Regular team events and offsites (~4x / year).
- Unlimited paid time off.
- Paid parental leave.
Tech Stack
PyTorch
Categories
AI & MLData Science