[ research ]

Research Engineer - Ai Systems

Yotta
Location
Remote
Work mode
Remote
Type
full time
Listed

Applications go directly to Yotta. Ramai takes no fee for placement and no money for ranking.

[ About this role ]

Research Engineer specializing in AI Systems at Yotta. This role focuses on optimizing AI workloads across heterogeneous hardware, including Trainium and GPUs. Responsibilities include developing high-performance kernels, optimizing inference runtimes, and designing scalable distributed training solutions.

Skills and focus areas

  • ai systems
  • compiler optimization
  • runtime optimization
  • distributed training
  • distributed inference
  • gpu kernel development
  • accelerator kernel development
  • large language model infrastructure
  • attention optimization
  • moe optimization
  • gemm optimization
  • collective communication optimization

Summary written by Ramai from the original posting. Full details and requirements are on Yotta’s careers page.

[ More at Yotta ]

Save money and win prizes up to $10m through weekly number draws.

See all 4 open roles at Yotta →

[ Similar roles at other startups ]

Browse all open research roles →

[ Not quite right? ]

Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.