{"id":69474,"url":"https://alion.io/job/nimble-seniorstaff-software-engineer-infrastructure-ml","title":"Senior/Staff Software Engineer, Infrastructure (ML)","company":{"id":30011,"name":"Nimble Robotics","domain":"nimble.ai","url":"https://alion.io/company/nimble","size_band":"51-200","is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Greenhouse","truth_index":{"grade":"B","score":82,"open_postings":13,"ghost_share":0,"stale_share":0.538,"repost_share":0,"time_to_fill_p50_days":65,"computed_at":"2026-10-12T06:55:09Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"staff","employment_type":null,"work_mode":"on_site","remote_scope":null,"remote_scope_basis":null,"remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["San Francisco, United States"],"countries":["US"],"hiring_countries":[],"hiring_countries_total":0,"salary":{"min":210000,"max":300000,"currency":"USD","period":"year","gross":null,"usd_annual":300000},"salary_estimate":null,"experience_years_min":4,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"C++","optional":false},{"name":"CUDA","optional":false},{"name":"CUDA Toolkit","optional":false},{"name":"JAX","optional":false},{"name":"Kubernetes","optional":false},{"name":"Python","optional":false},{"name":"PyTorch","optional":false},{"name":"PyTorch C++","optional":false},{"name":"Rust","optional":false},{"name":"DeepSpeed","optional":true},{"name":"FSDP","optional":true},{"name":"Megatron-LM","optional":true},{"name":"NCCL","optional":true}],"status":"live","first_seen_at":"2026-08-18T15:00:33Z","employer_posted_date":"2026-08-25","last_verified_at":"2026-10-12T10:39:09Z","board_verified":true,"closed_at":null,"days_open":54,"trust":{"level":"ok","repost_count":null,"flags":[],"days_open":54},"description":"About Nimble\nNimble is an AI robotics company building the autonomous supply chain to power fast, efficient and economical commerce. We’re training robot AGI to power a proprietary generalist supply chain superhumanoid, the first robot in the world capable of performing thousands of tasks across the supply chain. We’ve raised over $220M at over $1B valuation and formed a strategic alliance with FedEx to build a national network of autonomous warehouses capable of generating many billions in annual revenue. We are a hardcore and obsessed team of the world’s best engineers and operators. If you are obsessed with your craft, enjoy a high-intensity and fast moving high impact environment, are super high agency in getting hard things done and want to be part of building the world’s most legendary robotics company at the most pivotal moment in history, we want to work with you.\nWe are on a mission to empower and inspire mankind to accomplish legendary feats by inventing robots that liberate us from the menial. We will accomplish this by training robot AGI to invent and build the Autonomous Supply Chain - everything from the inside of factories and warehouses to your front door - powered by generalist superhumanoids.\nOur founding team comes from the AI labs at Stanford and Carnegie Mellon and our board of directors include famed robotics and AI legends including Fei-Fei Li (Chief Scientist of AI at Google and Director of Stanford’s AI Lab), Marc Raibert (founder of Boston Dynamics), and Sebastian Thrun (founder of GoogleX, Waymo; Stanford Professor and considered the father of autonomous vehicles).\nLet’s be legendary.\nWhy Join Nimble?\nAt Nimble, we are committed to building legendary products, a legendary team, and a legendary legacy. 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.\nNimble's Core Values:\nBe relentlessly resourceful - Challenge conventions and overcome obstacles.\nBe legendary - Be the very best and do work that inspires.\nBe humble - Prioritize growth, learning, and doing whatever is needed to further the mission.\nBe dependable - Take ownership and deliver with high agency.\nAbout the Role\nWe’re looking for a Software Engineer to join our ML Infrastructure team. In this role, you’ll help build the training and inference systems that power our general-purpose warehouse robots.\nYou’ll own training infrastructure end to end: keeping GPUs highly utilized, making runs reproducible, and ensuring every researcher can launch the next experiment with a single command. You’ll work closely with ML and Robotics teams to design, build, and scale the systems that turn our GPU clusters into a reliable, high-throughput platform for model development.\nResponsibilities\nDesign, develop, and maintain ML training infrastructure that enables the AI team to run training jobs efficiently, manage and iterate experiments quickly.\nBuild low-latency inference pipelines for production robotics workloads.\nDevelop, tune, and optimize low-level CUDA kernels.\nDesign training-platform systems for scalable model training, including high-throughput data ingestion, dataset sharding and sampling for distributed training.\nParticipate in and lead design reviews with peers and stakeholders to evaluate technical tradeoffs and select appropriate technologies.\nReview code and provide feedback to uphold best practices around style, correctness, testability, performance, and maintainability.\nContribute to documentation and educational materials, adapting content as systems and workflows evolve.\nMentor junior engineers and help raise the technical bar across the team.\nQualifications\nBachelor’s, Master’s, or PhD in Computer Science or a related field, or equivalent practical experience.\n4+ years of industry experience in infrastructure, distributed systems, ML systems, robotics, or a related area.\nExperience with programming languages such as Rust, Go, Python, or C++.\nExperience with ML frameworks such as PyTorch or JAX.\nStrong understanding of distributed systems, systems programming fundamentals, memory management, and performance optimization.\nExperience with Kubernetes orchestration, resource scheduling for large distributed jobs, and containerized deployment pipelines.\nAbility to debug and optimize bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operations.\nAbility to reason from first principles and optimize systems for both memory-bound and compute-bound workloads.\nStrong cross-functional communication skills, ownership, and a growth mindset.\nNice to Have\nHands-on experience with distributed training frameworks and techniques such as PyTorch DDP/FSDP, DeepSpeed, Megatron, or NCCL.\nHands-on experience with GPU kernel development.\nExperience with data engineering technologies such as Parquet, Arrow, or similar systems.\nCompensation\nThe pay range for this position at the start of employment is expected to bebetween $210,000 and $300,000/year. Your exact offer may vary depending on multiple individualized factors, including job-related knowledge, skills, and experience. In addition to cash compensation, this position will also receive generous equity for this position.\nCulture:\nWe 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 want to create something legendary and leave our mark on the world. 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!\nNimble Robotics, Inc. is an equal opportunity employer. We make all employment decisions based solely on merit. We provide equal employment opportunity to all applicants and employees without discrimination on the basis of race, color, religion, national origin, ancestry, disability, medical condition, genetic information, marital status, sex, gender, gender identity, gender expression, sexual orientation, age, military or veteran status, or any other characteristic protected by applicable state, federal or local laws.\nNimble's Benefits\nUnlimited Flexible Time Off\nEnjoy the time you need to travel, rejuvenate, and connect with friends and family.\nHealth Insurance\nNimble provides medical, dental, and vision insurance through several premier plans and options to support you and your family.\nPaid Parental Leave\nEnjoy paid bonding time following a birth.\nCommuter Benefits\nTake the stress out of commuting with access to fully-paid parking spots. \nReferral Bonus\nGet a cash bonus for any friend or colleagues that you refer to us that we end up hiring. \n401k\nContribute towards a 401k for retirement planning.\nEquity\nBe an owner in Nimble through our equity program\nLink: Nimble Closes $106 Million Series C Funding Round, Scales Fully Autonomous Fulfillment with FedEx \nLink: FedEx Announces Expansion of FedEx Fulfillment With Nimble Alliance","description_format":"text","description_chars":6996,"description_truncated":false,"requirements":{"experience_years_min":4,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":{"level":"bachelor","optional":true},"security_clearance":false,"languages":[]},"benefits":["401k plan","Equity","Flexible time off","Health insurance","Parental leave","Retirement plans","Vision insurance"],"hiring_locations":[],"hiring_excludes":[],"relocation_offered":false,"industries":["Artificial Intelligence","Logistics","Logistics & Warehouse Robotics"],"lifecycle":[{"event":"open","at":"2026-08-18T15:00:33Z"}],"visa":[{"country":"US","licensed_sponsor":true,"evidence":"Filed H-1B visas in the last two years","filings_12m":0,"filings_prev_12m":1,"green_card_filings_12m":0,"median_offered_wage_usd":null,"route":null,"cap_exempt":false,"checked_at":"2026-10-03T21:08:04+00:00","sources":["US Department of Labor: LCA disclosure data (H-1B, H-1B1, E-3)"],"filings_for_role_12m":0}],"liveness":{"score":45,"band":"ok","label":"Likely open","p_open":1,"p_active":0.595,"p_room":0.75,"age_days":54,"expected_fill_days":65,"reasons":["conf:2","stale_co","velocity","win:late"],"computed_at":"2026-10-12T06:55:09Z"},"pay":{"stated_usd_annual":300000,"is_top_pay":true},"html_url":"https://alion.io/job/nimble-seniorstaff-software-engineer-infrastructure-ml","json_url":"https://alion.io/job/nimble-seniorstaff-software-engineer-infrastructure-ml.json","meta":{"generated_at":"2026-10-12T11:45:44Z","cache_seconds":300,"methodology":"https://alion.io/methodology","terms":"https://alion.io/terms","contact":"https://alion.io/contact","api":"https://alion.io/developers","about":"Alion is a live layer of people, companies and AI agents: who they are, whether they are real and active right now, what they do and how to work with them, readable by people and by agents and paid per call.","catalog":"https://alion.io/catalog.json","usage":{"tier":"crawler_verified","counted_by":"address","units_charged":1,"used_today":18347,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-10-13T00:00:00Z"}},"offers":[{"id":"company.slices","title":"One company in depth, by slice","status":"live","price":{"credits":0.02,"usd":0.002,"plus_per_slice":{"credits":0.05,"usd":0.005}},"unit":"per company, plus each slice with data","note":"the employer in depth","call":{"mcp_tool":"get_company","arguments":{"id":30011},"rest":"https://alion.io/mcp/rest/get_company?id=30011"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fnimble-seniorstaff-software-engineer-infrastructure-ml"},{"id":"market.stats","title":"A market slice: pay, demand and time to fill","status":"live","price":{"credits":1,"usd":0.1},"unit":"per slice","note":"pay, demand and time to fill for this role and place","call":{"mcp_tool":"market_stats"},"human":"https://alion.io/catalog?offer=market.stats&for=job%2Fnimble-seniorstaff-software-engineer-infrastructure-ml"},{"id":"job.search","title":"Open jobs by role, technology, place, pay and visa","status":"live","price":{"credits":0.02,"usd":0.002},"unit":"per posting in a list","note":"similar open postings","call":{"mcp_tool":"search_jobs"},"human":"https://alion.io/catalog?offer=job.search&for=job%2Fnimble-seniorstaff-software-engineer-infrastructure-ml"},{"id":"company.verify","title":"Is this company real and active right now","status":"pilot","price":null,"unit":"per company","request":{"url":"https://alion.io/catalog/request","method":"POST","body":"{\"offer\": \"company.verify\", \"for\": \"job/nimble-seniorstaff-software-engineer-infrastructure-ml\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fnimble-seniorstaff-software-engineer-infrastructure-ml"}]}