Argmax
AI & Machine LearningDirect Greenhouse career site

On-device AI Frameworks Engineer (Intern)

Location
Palo Alto, CA · Hybrid
Job type
Internship
Level
Entry level
Listed
30 days ago

Skills & highlights for this role

  • iOS or Android SDK development
  • Swift or Kotlin with C++
  • mobile application profiling and optimization

Also useful

  • Core ML
  • MLX Swift
  • LiteRT
  • ONNX
  • WebGPU
  • ExecuTorch
  • Python

About the role

We are looking for an intern to join our 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.

AI 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.

Responsibilities

  • Contribute to SDK and Frameworks implementation: You will help implement new model support (example) and new features (example) as Argmax SDK grows. You will observe and learn from senior engineers at Argmax as they design the implementation spec that you will work on.
  • Respond to customer issues: You will be one of the first at Argmax to review customer issue reports, ask for supporting information, reproduce the issue, root-cause the issue, and finally propose viable workarounds as well as permanent fixes. Senior engineers at Argmax will support your growth as you learn to tackle this critical responsibility autonomously over time.

Qualifications

  • 3-6 months of hands-on experience in SDK or Frameworks development for iOS or Android
  • Familiarity with Swift OR (Kotlin AND C++)
  • Familiarity with profiling and optimizing mobile application

Preferred Qualifications

  • 1+ year of hands-on experience in SDK or Frameworks development for iOS or Android
  • Fluency in Swift, Kotlin and Python
  • Familiarity with at least one of Core ML, MLX Swift, LiteRT, ONNX, WebGPU or ExecuTorch

Perks

  • If internship is successful, this is an opportunity to join Argmax, a fast-growing early-stage startup with a unique mission
  • 3 days a week in the office from Palo Alto, CA or Manhattan, NY
  • Palo Alto office offers comprehensive on-site amenities, including chef-catered meals
  • Platinum-tier healthcare with 90% employer contribution, including dependents

About Argmax

AI applications are scaling in user adoption at unprecedented rates. The infrastructure is crumbling:

  • Spinner wheels are back in fashion
  • The most sensitive types of user data are uploaded to the cloud and occasionally leaked
  • Spiky demand leads to infrastructure capacity crunch and underutilization at the same time

Argmax is building the critical infrastructure required to bring real-time AI workloads to the edge:

  • Autoscaling instantly and infinitely
  • Private and compliant by design
  • Reliable beyond even the multi-cloud platforms

The 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!

Read the original posting

From Argmax's careers page, first listed on Rolev.ai on October 10, 2026. Rolev.ai is not the employer.

How Rolev.ai helps with this application

Rolev.ai is a job search and application tool for US job seekers. For this On-device AI Frameworks Engineer (Intern) role at Argmax, it shows how your resume fits, tailors your resume to the posting, and prepares the application for you to review.

  • See how you fit

    Openings from employer career sites are ranked against your whole resume—and searchable inside Claude, ChatGPT, and Cursor via Rolev.ai MCP.

  • Tailor without inventing

    Every change to your resume is checked against your original. New experience goes in only with your yes—zero fabricated claims.

  • Draft the full kit

    Tailored and Full Match resumes, role-specific cover letters, and answers to employer screening questions are prepared in seconds.

  • 1-Click Extension Autofill

    The Chrome extension fills the employer's Greenhouse, Workday, Lever, or Ashby form in 1 click. You review it and press Submit.

Start free with 5 tailored applications a monthChrome extensionFor AI assistantsPricingHow Rolev.ai comparesHow Greenhouse screens applications

Similar roles

All Software Developer jobs

Common questions

Where is the On-device AI Frameworks Engineer (Intern) job at Argmax located?

This On-device AI Frameworks Engineer (Intern) position at Argmax is located in Palo Alto, CA (Hybrid). Always check Argmax's application page for specific state eligibility or relocation policies.

What qualifications are required for On-device AI Frameworks Engineer (Intern) at Argmax?

For the On-device AI Frameworks Engineer (Intern) role, Argmax highlights experience with iOS or Android SDK development, Swift or Kotlin with C++, mobile application profiling and optimization. Rolev.ai can score your existing resume against these requirements and show any skill gaps before you apply.

How do I apply for the On-device AI Frameworks Engineer (Intern) position at Argmax?

This listing links directly to Argmax's Greenhouse application page. You can apply directly on Argmax's site, or use Rolev.ai to check your resume fit, generate a zero-hallucination tailored resume and cover letter for this posting, and autofill the Greenhouse application form in 1 click with the Rolev.ai Chrome extension. You always review the form and press Submit yourself.

Tailor resume & applyCompany site