Staff Software Engineer - ML Infrastructure
- Location
- San Francisco, CA · On-site
- Job type
- Full-time
- Level
- Lead / Staff
- Listed
- 30 days ago
Skills & highlights for this role
- distributed training systems
- PyTorch
- TensorFlow
- GPU performance optimization
- Python
About the Role
At Watney, ML Infrastructure engineers turn data collected from a live fleet of robots into better models. The fleet produces large volumes of video and telemetry data from real work in the field, and making that data trainable is one of the hardest systems problems at the company.
As we continue to scale, these systems will require larger training runs with more data, expanded clusters, and optimal GPU utilization.
What You’ll Do
- Own training and inference infrastructure.
- Build the data pipelines that these training runs depend on.
- Make experiments fast to launch and reproduce.
- Contribute to our core training code.
You May Be a Good Fit If You
- Have built ML infrastructure that carried real production training runs.
- Have scaled distributed training systems.
- Strong experience with Python, PyTorch or TensorFlow.
- Have experience identifying and troubleshooting GPU performance bottlenecks in large-scale training environments.
Additional information
We’re committed to building a diverse, inclusive team. At Watney Robotics, we welcome people of all backgrounds and identities, and we make hiring decisions based on skills, experience, and potential. If you’re passionate about robotics but don’t meet every requirement, we still encourage you to apply!
CURIOUS TO LEARN MORE?
Our Mission
Expand human ambition in the physical world.
Critical infrastructure is constrained by labor shortages, hazardous working conditions, and operational complexity. Watney builds and deploys autonomous robotic systems that increase the speed and capacity of buildout, starting with data centers.
From Watney's careers page, first listed on Rolev.ai on October 10, 2026. Rolev.ai is not the employer.
How Rolev.ai helps with this application
Rolev.ai is a job search and application tool for US job seekers. For this Staff Software Engineer - ML Infrastructure role at Watney, it shows how your resume fits, tailors your resume to the posting, and prepares the application for you to review.
See how you fit
Openings from employer career sites are ranked against your whole resume—and searchable inside Claude, ChatGPT, and Cursor via Rolev.ai MCP.
Tailor without inventing
Every change to your resume is checked against your original. New experience goes in only with your yes—zero fabricated claims.
Draft the full kit
Tailored and Full Match resumes, role-specific cover letters, and answers to employer screening questions are prepared in seconds.
1-Click Extension Autofill
The Chrome extension fills the employer's Greenhouse, Workday, Lever, or Ashby form in 1 click. You review it and press Submit.
Similar roles
All Software Developer jobs- Software DeveloperAccenture Federal Services · Washington, DC · $97.5k – $115k/yr
- Software Developer 803Freedom Technology Solutions Group · Annapolis Junction, MD · $150k – $200k/yr
- Software DeveloperPurvis Systems · Newport, RI
- NASA Human Spaceflight Artificial Intelligence (AI) Software Engineer and Data ScientistCACI · Houston, TX
- Senior Manager Software EngineeringNasdaq · Atlanta, GA
- Software Engineering Supervisor - Test SystemsMoog · Torrance, CA
Common questions
Where is the Staff Software Engineer - ML Infrastructure job at Watney located?
This Staff Software Engineer - ML Infrastructure position at Watney is located in San Francisco, CA (On-site). Always check Watney's application page for specific state eligibility or relocation policies.
What qualifications are required for Staff Software Engineer - ML Infrastructure at Watney?
For the Staff Software Engineer - ML Infrastructure role, Watney highlights experience with distributed training systems, PyTorch, TensorFlow, GPU performance optimization, Python. Rolev.ai can score your existing resume against these requirements and show any skill gaps before you apply.
How do I apply for the Staff Software Engineer - ML Infrastructure position at Watney?
This listing links directly to Watney's Ashby application page. You can apply directly on Watney's site, or use Rolev.ai to check your resume fit, generate a zero-hallucination tailored resume and cover letter for this posting, and autofill the Ashby application form in 1 click with the Rolev.ai Chrome extension. You always review the form and press Submit yourself.