Member of Technical Staff, AI Research

Added
19 hours ago
Type
Full time
Salary
Salary not provided

Related skills

pytorch reinforcement learning jax ray accelerate

๐Ÿ“‹ Description

  • Build and train AI agents and training systems to learn physics.
  • Design evaluation workflows and benchmarks for physics reasoning.
  • Publish results; develop curricula, rewards, and architectures for discovery tasks.
  • Collaborate with physicists and engineers; ship end-to-end systems.

๐ŸŽฏ Requirements

  • PhD in ML, CS, physics, math with top-venue publications.
  • Hands-on RL experience building agents and training models.
  • Proficiency with ML frameworks and distributed training; scale-ready.
  • Physics or math background for physical reasoning and tool use.
  • Hands-on RL algorithms such as PPO, SAC, MuZero.
  • PyTorch or JAX; distributed training via Ray, XLA, Accelerate.
  • Experience applying agents to simulators or scientific tools.
  • Open-source contributions or shipped artifacts.

๐ŸŽ Benefits

  • Competitive salary and benefits.
  • Meaningful early-stage equity.
  • Equal opportunity employer; diverse perspectives valued.
Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest โ€” finding, filtering, and applying while you focus on what matters.

Related Engineering Jobs

See more Engineering jobs โ†’