Senior Staff Software Engineer Infrastructure ML
Nimble Robotics operates fully autonomous e-commerce fulfillment centers and a robotic 3PL service powered by general-purpose warehouse robots and cloud logistics software.
Funding history
About Nimble Robotics
Nimble develops AI-powered general-purpose warehouse robots that perform storage, retrieval, picking, packing, sorting, and related fulfillment tasks. It combines the robots with its Cloud Logistics Platform to operate end-to-end autonomous fulfillment centers for e-commerce brands.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will build and optimize ML training and inference systems for warehouse robotics. You will improve GPU utilization, reproducibility, distributed training, data ingestion, and low-latency inference. You will develop CUDA kernels, review code, document systems, and mentor junior engineers.
Requirements
- Infrastructure, distributed systems, ML systems, or robotics experience
- Rust, Go, Python, or C++
- PyTorch or JAX
- Distributed system
- Systems programming
- Memory management
- Performance optimization
- Kubernetes
- Resource scheduling
- Containerized deployment
- GPU memory
- Networking
- Filesystem
- Multi-GPU operation
Responsibilities
- Design, develop, and maintain ML training infrastructure
- Build low-latency inference pipelines for production robotics workloads
- Develop, tune, and optimize CUDA kernels
- Design scalable training-platform systems, data ingestion, dataset sharding, and sampling
- Lead design reviews and evaluate technical tradeoffs
- Review code for correctness, testability, performance, and maintainability
- Create and maintain technical documentation and educational materials
- Mentor junior engineers
Benefits
- Unlimited flexible time off
- Medical, dental, and vision insurance
- Paid parental leave
- Commuter benefits with fully paid parking
- 401k
- Equity
