Research Engineer

Requirements

• PhD in CS, Math, or AI
• Advanced knowledge of PyTorch
• Knowledge of AI research codebases and workflows

Nice to Haves

• Expertise implementing CUDA or Triton kernels

What You'll Be Doing

• Tech lead a state of the art research codebase on large language models, world models, reinforcement learning
• Full-stack development from integrating with kernel specialists to research trainings at scale
• Cross-org leadership (e.g. PyTorch)

Perks and Benefits

• No specific perks and benefits listed