Software Engineer - ML Infrastructure
Watney Robotics Inc
- Location
- San Francisco, California, United States
- Job type
- Full-time
Required skills
- JAX
- PyTorch
- FSDP
- DeepSpeed
- Megatron
- Ray Train
- vLLM
- Triton
- Ray Serve
- Python
- Rust
- C/C++
About the role
Experience building ML platforms and large-scale distributed training, fluency in Python and Rust or C/C++, familiarity with JAX and PyTorch, knowledge of distributed backbones (FSDP, DeepSpeed, Megatron, Ray Train), GPU/TPU optimization, and cluster orchestration.
About Watney Robotics Inc
Develops autonomous robotic systems for data center infrastructure buildout.
This page is fully interactive when JavaScript is enabled. Please enable JavaScript to apply or browse related roles.
