fsdp

Recent positions

• Principal Machine Learning Engineer

  bjakcareer · 24d ago

  principal · remote · full time · London

  pytorch jax ray fsdp deepspeed megatron +1
  Apply

• Senior AI Infrastructure Engineer - Training Platform

  scaleai · 26d ago

  $216k – $270k

  senior · on-site · full time · New York, San Francisco

  c++ python pytorch cuda aws terraform +13
  Apply

• Research Engineer, Video Generation

  mirage · Apr 14, 2026

  $175k – $275k

  mid · on-site · full time · New York

  pytorch cuda fsdp triton
  Apply

• Principal Machine Learning Engineer

  bjakcareer · Apr 11, 2026

  $180k – $250k

  senior · remote · full time · Singapore

  pytorch jax spark ray fsdp deepspeed +4
  Apply

• Machine Learning Engineer - Distributed ML Systems

  pluralis-research · Apr 1, 2026

  senior · remote · full time · San Francisco

  python grpc fsdp deepspeed megatron
  Apply

• Research Engineer - RL Infrastructure

  PrimeIntellect · Mar 27, 2026

  $150k – $300k

  senior · on-site · full time · San Francisco

  pytorch cuda vllm rl open-source gpu +11
  Apply

• Senior Cloud Infrastructure Engineer

  gatikaiinc · Mar 18, 2026

  $180k – $240k

  senior · on-site · full time · Mountain View

  terraform kubernetes opentelemetry grafana kafka prometheus +10
  Apply

• Research Engineer, Multimodal

  character · Mar 6, 2026

  $225k – $400k

  senior · remote · full time · Redwood City

  pytorch docker kubernetes slurm fsdp deepspeed
  Apply

• Machine Learning Engineer - Distributed ML Systems

  pluralis-research · Feb 23, 2026

  senior · remote · full time · Melbourne

  python grpc fsdp deepspeed megatron
  Apply

• Scientist /Senior Scientist, Multimodal & Relational Machine Learning Foundation Models

  altoslabs · Feb 17, 2026

  $200k – $257k

  senior · on-site · full time · Redwood City

  python pytorch jax ray fsdp deepspeed +2
  Apply

• Principal Machine Learning Engineer

  bjakcareer · Feb 11, 2026

  $180k – $250k

  principal · remote · full time · Beijing

  pytorch vllm jax spark ray fsdp +8
  Apply

• Principal Machine Learning Engineer

  bjakcareer · Feb 11, 2026

  $180k – $250k

  principal · remote · full time

  pytorch jax ray fsdp deepspeed megatron +1
  Apply

• ML Infrastructure Engineer

  echo · Jan 29, 2026

  $180k – $230k

  senior · remote · full time · San Francisco

  c++ python pytorch cuda docker kubernetes +7
  Apply

• Research Engineer, Machine Learning

  mistral · Jan 27, 2026

  $180k – $200k

  senior · on-site · full time · Palo Alto

  pytorch tensorflow jax k8s slurm fsdp +1
  Apply

• Principal Engineer - Perf and Benchmarking

  coreweave · Dec 18, 2025

  $206k – $333k

  principal · on-site · full time · Sunnyvale

  tensorrt cuda aws kubernetes nvidia grafana +15
  Apply

• Research Scientist - Model Team

  mirelo · Dec 3, 2025

  senior · remote · full time · Berlin

  pytorch autoregressive transformer fsdp diffusion low precision training +1
  Apply

• Machine Learning Engineer/Machine Learning Scientist , Multi Modality

  altoslabs · Nov 24, 2025

  $212k – $273k

  senior · on-site · full time · San Diego

  python pytorch tensorflow jax ray fsdp +3
  Apply

• Research Engineer, Frontier Speculative Decoding

  togetherai · Nov 20, 2025

  $190k – $270k

  mid · on-site · full time · San Francisco

  python pytorch kubernetes slurm fsdp deepspeed
  Apply

• Machine Learning - Infrastructure

  causal · Oct 29, 2025

  $100k – $2000k

  senior · on-site · full time · San Francisco

  docker kubernetes gcp fsdp deepspeed
  Apply

• Distributed Training Engineer

  periodic-labs · Sep 24, 2025

  senior · on-site · full time · Menlo Park

  fsdp deepspeed megatron-lm
  Apply