{"id":104547,"url":"https://alion.io/job/waabi-senior-staff-ml-training-optimization-engineer","title":"Senior / Staff ML Training Optimization Engineer","company":{"id":4483,"name":"Waabi","domain":"waabi.ai","url":"https://alion.io/company/waabi","size_band":"1001-5000","is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Lever","truth_index":{"grade":"A","score":92,"open_postings":66,"ghost_share":0,"stale_share":0.106,"repost_share":0,"time_to_fill_p50_days":150,"computed_at":"2026-10-17T05:46:00Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"staff","employment_type":"full_time","work_mode":"remote","remote_scope":"stated_countries","remote_scope_basis":"board_field","remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["San Francisco, United States","Dallas, United States","Pittsburgh, United States","Phoenix, United States","Toronto, Canada","Canada"],"countries":["US","CA"],"hiring_countries":["US","CA"],"hiring_countries_total":2,"salary":null,"salary_estimate":{"min_usd":256000,"max_usd":522000,"period":"year","method":"role_seniority_country_remote_cell","sample_n":823},"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":"Python","optional":false},{"name":"PyTorch","optional":false},{"name":"PyTorch C++","optional":false},{"name":"Quantization","optional":false},{"name":"Rust","optional":false},{"name":"Bazel","optional":true},{"name":"Kubernetes","optional":true}],"status":"live","first_seen_at":"2026-05-08T16:51:32Z","employer_posted_date":"2026-05-08","last_verified_at":"2026-10-17T20:09:44Z","board_verified":true,"closed_at":null,"days_open":162,"trust":{"level":"ok","repost_count":0,"flags":[],"days_open":161},"description":"You will...\nBuild standardized distributed training frameworks for research and production, drive our training towards new levels of stability and efficiency.\nComprehensively profile model runtime and memory to pinpoint performance bottlenecks.\nIdentify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks. Examples include designing new CUDA kernels, quantization-aware training and inference, and compilation/deployment techniques.\nWork with researchers and ML engineers on best-practices for optimal resource usage.\nCreate and improve tooling and dashboards to ensure broad adoption of your work.\nQualifications:\nMS/PhD or Bachelors degree with a minimum of 4 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study.\nSolid coding proficiency in a variety of coding languages including Python, C++ or Rust.\nExperience in deep learning frameworks such as PyTorch or Jax.\nSkilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.\nOpen-minded and collaborative team player with willingness to help others.\nPassionate about self-driving technologies, solving hard problems, and creating innovative solutions.\nBonus/nice to have:\nExperience in identifying when custom CUDA kernels are needed, and implementing them.\nExperience in Bazel in a monorepo environment, and integrating third party packages into dev environments.\nExperience with Kubernetes-based training platforms.","description_format":"text","description_chars":1504,"description_truncated":false,"requirements":{"experience_years_min":4,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":{"level":"bachelor","optional":false},"security_clearance":false,"languages":[]},"benefits":[],"hiring_locations":[{"name":"United States","iso":"US","kind":"country"},{"name":"Canada","iso":"CA","kind":"country"}],"hiring_excludes":[],"relocation_offered":false,"industries":["Autonomous Driving"],"lifecycle":[{"event":"open","at":"2026-05-08T16:51:32Z"}],"visa":[{"country":"US","licensed_sponsor":true,"evidence":"H-1B filings in 12 months: 9","filings_12m":9,"filings_prev_12m":0,"green_card_filings_12m":0,"median_offered_wage_usd":172500,"route":null,"cap_exempt":false,"checked_at":"2026-10-15T11:24:14+00:00","sources":["US Department of Labor: LCA disclosure data (H-1B, H-1B1, E-3)"],"filings_for_role_12m":6}],"liveness":{"score":30,"band":"fade","label":"Fading","p_open":1,"p_active":0.673,"p_room":0.44,"age_days":161,"expected_fill_days":150,"reasons":["conf:2","velocity","win:tail","crowd:"],"computed_at":"2026-10-17T05:46:00Z"},"pay":null,"html_url":"https://alion.io/job/waabi-senior-staff-ml-training-optimization-engineer","json_url":"https://alion.io/job/waabi-senior-staff-ml-training-optimization-engineer.json","meta":{"generated_at":"2026-10-17T22:12:37Z","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":7239,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-10-18T00: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":4483},"rest":"https://alion.io/mcp/rest/get_company?id=4483"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fwaabi-senior-staff-ml-training-optimization-engineer"},{"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%2Fwaabi-senior-staff-ml-training-optimization-engineer"},{"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%2Fwaabi-senior-staff-ml-training-optimization-engineer"},{"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/waabi-senior-staff-ml-training-optimization-engineer\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fwaabi-senior-staff-ml-training-optimization-engineer"}]}