1 day ago
San Francisco, CA, USA
Mid Level / Senior
H1B Sponsor
Base Salary
$350k - $850k/yr
Responsibilities
- Invent, design, and implement reinforcement learning environments and evaluations.
- Conduct experiments and shape the research roadmap.
- Deliver work into training runs.
- Collaborate with researchers, engineers, and performance engineering specialists.
Requirements
- Expertise with accelerators such as CUDA, ROCm, Triton, or Pallas.
- Proficiency in ML framework programming with JAX or PyTorch.
- Experience working across the tech stack, including kernels and distributed systems.
- Ability to balance research exploration with engineering implementation.
- Passion for AI and commitment to developing safe and beneficial systems.
Benefits
- Competitive compensation and benefits.
- Optional equity donation matching.
- Generous vacation and parental leave.
- Flexible working hours.
- Collaborative office space.
Tech Stack
PyTorch
Categories
AI & MLData Science