Nimble Robotics
Senior/Staff Software Engineer, Infrastructure (ML)
San Francisco, CA · Staff+
Sponsorship not specifiedDetected 58 days ago
PythonGoRustDistributed SystemsKubernetesMachine LearningPyTorchData EngineeringValuationSupply ChainRoboticsCommunication
Stay score
odds of building a lasting career here
37Risky
Cap-exempt (no lottery)0
Sponsors this role80
Entry-level history0
PERM / green-card track0
Lottery odds40
Fits your clock70
Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.
Lottery odds assume a STEM candidate.
Personalize to your clock →Employer immigration record
from this employer's Department of Labor filings
Files H-1B transfers
9 transfer filings in the last year, covering 9 workers. Median labor-condition decision: 8 days. An employer that already files transfers is one that can take over an existing H-1B.
Sourced from Department of Labor LCA, PERM and prevailing-wage disclosure data. Employer matching is by name, so figures may be split across an employer's legal entities. Absence of a filing means none appears in our copy of the data, not that none exists.
Community outcomes
No reports yet — be the first to help the next applicant.
About the role
- We're looking for a Software Engineer to join our ML Infrastructure team.
- keeping GPUs highly utilized, making runs reproducible, and ensuring every researcher can launch the next experiment with a single command.
Responsibilities
- Design, develop, and maintain ML training infrastructure that enables the AI team to run training jobs efficiently, manage and iterate experiments quickly.
- Build low-latency inference pipelines for production robotics workloads.
- Develop, tune, and optimize low-level CUDA kernels.
- Design training-platform systems for scalable model training, including high-throughput data ingestion, dataset sharding and sampling for distributed training.
- Participate in and lead design reviews with peers and stakeholders to evaluate technical tradeoffs and select appropriate technologies.
- Nimble is an AI robotics company building the autonomous supply chain to power fast, efficient and economical commerce.
Requirements
- Bachelor's, Master's, or PhD in Computer Science or a related field, or equivalent practical experience.
- 4+ years of industry experience in infrastructure, distributed systems, ML systems, robotics, or a related area.
- Experience with programming languages such as Rust, Go, Python, or C++.
- Experience with ML frameworks such as PyTorch or JAX.
- Strong understanding of distributed systems, systems programming fundamentals, memory management, and performance optimization.
- Experience with Kubernetes orchestration, resource scheduling for large distributed jobs, and containerized deployment pipelines.
- Ability to debug and optimize bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operations.
- Ability to reason from first principles and optimize systems for both memory-bound and compute-bound workloads.
- Strong cross-functional communication skills, ownership, and a growth mindset.
Nice to have
- Hands-on experience with distributed training frameworks and techniques such as PyTorch DDP/FSDP, DeepSpeed, Megatron, or NCCL.
- Hands-on experience with GPU kernel development.
- Experience with data engineering technologies such as Parquet, Arrow, or similar systems.
- Your exact offer may vary depending on multiple individualized factors, including job-related knowledge, skills, and experience.
- We embrace challenges and strive to make the impossible possible each day.
- We're not in this to do what's easy or to be mediocre.
- We're ambitious, we're gritty, we're humble and we're relentlessly resourceful in pursuit of our goals.
- If this sounds like you then you might be a great fit!
Compensation
- The pay range for this position at the start of employment is expected to be between $210,000 and $300,000/year.
- In addition to cash compensation, this position will also receive generous equity for this position.
Benefits
- Contribute to documentation and educational materials, adapting content as systems and workflows evolve.
- We want to create something legendary and leave our mark on the world.
- Nimble's Benefits
- Unlimited Flexible Time Off
- Health Insurance
- Nimble provides medical, dental, and vision insurance through several premier plans and options to support you and your family.
- Paid Parental Leave
- Commuter Benefits
- Referral Bonus
Company info
- We are a hardcore and obsessed team of the world's best engineers and operators.
- We are on a mission to empower and inspire mankind to accomplish legendary feats by inventing robots that liberate us from the menial.
- Join us and become part of an ambitious, humble, and resourceful culture where your work will leave a lasting impact on the future of robotics and commerce.
Apply directly at Nimble Robotics →Create a free account for alerts like thisView Nimble Robotics immigration profile
This listing is sourced directly from Nimble Robotics's careers page and normalized into a canonical job model.