{"id":66218,"url":"https://alion.io/job/argmax-on-device-ai-frameworks-engineer-staff","title":"On-device AI Frameworks Engineer (Staff)","company":{"id":33592,"name":"Argmax","domain":"argmax.org","url":"https://alion.io/company/argmax","size_band":null,"is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Greenhouse","truth_index":{"grade":"B","score":75,"open_postings":3,"ghost_share":0,"stale_share":1,"repost_share":0,"time_to_fill_p50_days":null,"computed_at":"2026-10-17T05:46:00Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"staff","employment_type":null,"work_mode":"hybrid","remote_scope":null,"remote_scope_basis":null,"remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["Palo Alto, United States"],"countries":["US"],"hiring_countries":[],"hiring_countries_total":0,"salary":null,"salary_estimate":{"min_usd":186000,"max_usd":380000,"period":"year","method":"role_seniority_country_remote_cell","sample_n":823},"experience_years_min":5,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"Core ML","optional":false},{"name":"Edge AI","optional":false},{"name":"ExecuTorch","optional":false},{"name":"LiteRT","optional":false},{"name":"MLX ML","optional":false},{"name":"ONNX","optional":false},{"name":"Python","optional":false},{"name":"Swift","optional":false},{"name":"WebGPU","optional":false},{"name":"JavaScript","optional":true},{"name":"Kotlin","optional":true}],"status":"live","first_seen_at":"2025-10-29T20:46:16Z","employer_posted_date":"2025-11-05","last_verified_at":"2026-10-18T01:25:25Z","board_verified":true,"closed_at":null,"days_open":353,"trust":{"level":"stale","repost_count":0,"flags":["stale"],"days_open":352},"description":"We are looking for a Staff Engineer to join our growing On-device AI Frameworks team at Argmax! In this role, you will design, implement and optimize software frameworks that expose developer-friendly APIs to run state-of-the-art inference workloads natively on Apple and Android devices. You will collaborate closely with industry-leader engineer and researcher colleagues, advancing the frontiers of on-device inference technology and accelerating its market adoption.\nAI Frameworks are at the core of Argmax SDK, our flagship developer toolkit trusted by Enterprises and developers in high-stakes industries such as healthcare. Argmax is a customer-obsessed team and we work very closely with them, sometimes in forward-deployed capacity.\nResponsibilities\n Productionize research prototypes: The Applied Research team will come to you with a Python prototype that provides the blueprints for a new feature (example) for one of our AI Frameworks. You will collaborate with them in turning this standalone prototype into a production-ready implementation in a test-driven development fashion. This collaboration may include experimentation and benchmarking that leads to top-tier AI research conference submissions (example). \nContribute to SDK and Frameworks design: As the number of supported models and workloads grow, you will see around corners and recommend scalable design patterns to absorb the code growth while minimizing technical debt.\nSupport Enterprise customers: You will directly work with the engineering teams of Enterprise customers during their onboarding journey. This could range from a Q&A session to customizing an Argmax feature to fit a particular customer requirement.\nQualifications\n3+ years of hands-on experience in SDK or Frameworks development for iOS or macOS\nFluency in Swift\nFluency in profiling and optimizing native applications\nFamiliarity with at least one of Core ML, MLX Swift, LiteRT, ONNX, WebGPU or ExecuTorch\nPreferred Qualifications\n5+ years of hands-on experience in SDK or Frameworks development for iOS or Android\nTrack record of significant open-source contributions\nFluency in Swift, Kotlin and Python\nDirect experience with Core ML, MLX Swift and LiteRT\nPerks\nTop-of-market equity at a fast-growing early-stage startup with a unique mission\nPerformance-based equity refreshers twice a year\n3 days a week in the office from Palo Alto, CA or Manhattan, NY\nPalo Alto office offers comprehensive on-site amenities, including chef-catered meals\nRemote possible by exception for industry leader exceptional candidates\nPlatinum-tier healthcare with 90% employer contribution, including dependents\n401(k) match\nQuarterly in-person team-building weeks in Palo Alto, CA\nAbout Argmax\nAI applications are scaling in user adoption at unprecedented rates. The infrastructure is crumbling:\n- Spinner wheels are back in fashion\n- The most sensitive types of user data are uploaded to the cloud and occasionally leaked\n- Spiky demand leads to infrastructure capacity crunch and underutilization at the same time\nArgmax is building the critical infrastructure required to bring real-time AI workloads to the edge:\n- Autoscaling instantly and infinitely\n- Private and compliant by design\n- Reliable beyond even the multi-cloud platforms\nThe hardest part: We are directly migrating cloud workloads to the edge, without compromising accuracy that our customers work so hard to achieve. This is a hard core technology problem and we built the mission-driven team with a long-term vision to make on-device the default way to build AI applications. Join us if this sounds like you and you have 3+ years to stake!","description_format":"text","description_chars":3641,"description_truncated":false,"requirements":{"experience_years_min":5,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":null,"security_clearance":false,"languages":[]},"benefits":["Equity"],"hiring_locations":[{"name":"United States","iso":"US","kind":"country"}],"hiring_excludes":[],"relocation_offered":false,"industries":["Machine Learning","AI Research Labs"],"lifecycle":[{"event":"open","at":"2025-10-29T20:46:16Z"}],"visa":[{"country":"US","licensed_sponsor":true,"evidence":"H-1B filings in 12 months: 1","filings_12m":1,"filings_prev_12m":3,"green_card_filings_12m":0,"median_offered_wage_usd":142500,"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":1}],"liveness":{"score":9,"band":"cold","label":"Long shot","p_open":1,"p_active":0.319,"p_room":0.28,"age_days":352,"expected_fill_days":30,"reasons":["conf:3","win:tail","crowd:"],"computed_at":"2026-10-17T05:46:00Z"},"pay":null,"html_url":"https://alion.io/job/argmax-on-device-ai-frameworks-engineer-staff","json_url":"https://alion.io/job/argmax-on-device-ai-frameworks-engineer-staff.json","meta":{"generated_at":"2026-10-18T03:32:48Z","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":5545,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-10-19T00: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":33592},"rest":"https://alion.io/mcp/rest/get_company?id=33592"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fargmax-on-device-ai-frameworks-engineer-staff"},{"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%2Fargmax-on-device-ai-frameworks-engineer-staff"},{"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%2Fargmax-on-device-ai-frameworks-engineer-staff"},{"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/argmax-on-device-ai-frameworks-engineer-staff\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fargmax-on-device-ai-frameworks-engineer-staff"}]}