Performance RL Research Engineer: Safe Code for Accelerators

Menlo Ventures; San Francisco (CA); Hybrid; USD 350,000 - 850,000; Full time

Benefits offered by this job

Competitive compensation

Generous vacation and parental leave

Flexible working hours

Job summary

Menlo Ventures is looking for a Research Engineer for their Code RL team. This role focuses on advancing AI models' coding capabilities while ensuring safety and performance in output.

The successful candidate should have expertise in machine learning frameworks and a Bachelor's degree or equivalent experience. The position offers competitive compensation, including a salary range of $350,000 to $850,000, and benefits including flexible working hours and a lovely office in San Francisco.

Qualifications

• Expertise with CUDA, ROCm, or other accelerators.
• Experience with ML frameworks like JAX or PyTorch.
• Ability to balance research and engineering.

Responsibilities

• Invent and implement RL environments and evaluations.
• Conduct experiments and shape the research roadmap.
• Deliver work into training runs.
• Collaborate with researchers and engineers across teams.

Skills

Expertise with accelerators (CUDA, ROCm, Triton, Pallas)

ML framework programming (JAX or PyTorch)

Experience with reinforcement learning

Knowledge of distributed systems

Education

Bachelor's degree in a related field

Job description

Menlo Ventures is looking for a Research Engineer for their Code RL team. This role focuses on advancing AI models' coding capabilities while ensuring safety and performance in output.

The successful candidate should have expertise in machine learning frameworks and a Bachelor's degree or equivalent experience. The position offers competitive compensation, including a salary range of $350,000 to $850,000, and benefits including flexible working hours and a lovely office in San Francisco.