On-device AI Frameworks Engineer (Intern)
Argmax
Overview 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 leaders, advancing the frontiers of on-device inference technology. Responsibilities Contribute to SDK and Frameworks implementation: help implement new model support and features as Argmax SDK grows. Respond to customer issues: review issue reports, reproduce problems, root‑cause them, and propose workarounds or fixes. 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 applications. 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. If internship is successful, opportunity to join Argmax. 3 days a week in the office from Palo Alto, CA or Manhattan, NY. Palo Alto office offers comprehensive on‑site amenities. 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, and we are building critical infrastructure to bring real‑time AI workloads to the edge. Our mission is to make on‑device the default way to build AI applications. #J-18808-Ljbffr Argmax
- 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...SuggestedWork at officeRemote work3 days per week
- Argmax, Inc. in Palo Alto, CA is looking for a Staff Engineer for its On-device AI Frameworks team. This role focuses on designing and optimizing software frameworks for native inference workloads on both Apple and Android devices. You will work closely with research teams...SuggestedWork at office
- Argmax seeks a Staff Engineer for its On-device AI Frameworks team in Palo Alto, CA. In this role, you will design, implement, and optimize software frameworks for native AI workloads on Apple and Android devices. Collaborate closely with research teams, turning prototypes...SuggestedFlexible hours
- Argmax is looking for an intern to join their On-device AI Frameworks team in Palo Alto, CA. The role involves designing, implementing, and optimizing software frameworks for mobile devices. You'll contribute to SDK development and respond to customer issues, all while...Internship
- Google Inc. is seeking a Senior Software Engineer for the AI Core Solutions team in Mountain View, CA. This role involves developing mobile AI innovations and optimizing on-device GenAI capabilities. Ideal candidates have a Bachelor's degree and substantial experience...Suggested
- Rivian and Volkswagen Group Technologies is seeking an AI Engineer Intern for hands-on experience in developing speech, vision, and multimodal... ...field, has strong Python skills, and is familiar with AI frameworks. In addition to a competitive salary, the position includes...InternshipFull time
- Majestic Labs is looking for an Embedded Systems Software Engineer to design and optimize AI-based embedded systems. The ideal candidate will work with cross-functional teams to enhance smart devices, involving development and integration of AI solutions. Successful candidates...
$25 - $65 per hour
Tesla Motors, Inc. is seeking an AI/ML Modeling Engineering Intern to join their AI Hardware team in Palo Alto. As an intern, you will collaborate... ...requires deep expertise in computer architecture and ML frameworks. The ideal candidate is pursuing a degree in related fields...InternshipHourly pay$100k - $150k
...positions, not internships. International Students: If your work authorization... ...future generations of Tesla AI products. You will work on a... ...clusters with a data engine that constantly generates new... ...experience in machine learning frameworks and models such as PyTorch, TensorFlow...InternshipFull timeTemporary workPart timeFlexible hours$50 - $65 per hour
Internship, Applied AI Engineer, AI Hardware (Fall 2026) Consider before submitting an application... ...IC flows Expertise in Python + ML frameworks (PyTorch, TensorFlow, JAX); experience... ...Compensation and Benefits As a full‑time Tesla Intern, you will be eligible for: Medical...InternshipFull timeTemporary workSummer workFlexible hours- ...Presto is the leading Voice AI company for restaurant drive‑... ...deeply technical, entrepreneurial engineering culture. We place a premium... ...for Fall 2026 AI Engineering Interns to contribute to the core... ...systems. Familiarity with ML frameworks such as PyTorch or TensorFlow...InternshipFull time
$20 - $50 per hour
...full-time positions, not internships. International Students: If your work authorization is... ...solar, battery storage, and car charging devices in many countries around the world. And... ...team is looking for a Software Engineer to build control and configuration software...InternshipFull timeTemporary workPart timeLocal areaFlexible hours$174k - $253k
Google Inc. is seeking a Senior Software Engineer for its Sensor AI/ML/Watch Software team in Mountain View, CA. This role involves designing AI... ...for sensor fusion, managing model optimization for edge devices, and developing high-performance sensor pipelines in C/C++...- ...AI/ML Engineering Intern Company: Synopsys Location: Mountain View, CA (On-site, Remote) Category: Interns/Temp Hire Type: Intern Job ID: 17595 Base Salary Range: $0-$0 Remote Eligible: Yes Date Posted: 05/27/2026 About This Opportunity Synopsys' Generative AI Center...InternshipFull timeTemporary workPart timeSummer workSummer internshipWork at officeRemote work
$30 - $59 per hour
d-Matrix is seeking an Applied AI Engineering Intern for its Intelligent Manufacturing Systems team in Santa Clara, CA. This 12-week summer... ...enhance manufacturing processes, utilizing data analysis and ML frameworks. The ideal candidate is pursuing a Master's or PhD in a...InternshipHourly paySummer internship- ...AI Software Engineering Intern CSBio is a leading biotech company in the peptide industry, and for over 30 years has been providing pharmaceutical instrumentation to biotech, pharmaceutical, and academic researchers. We are looking for a highly motivated AI Software...InternshipFull timeWork at officeLocal areaImmediate start
$7,000 - $8,000 per month
Deep Infra Inc. is seeking a talented Software Engineering Intern to work with an experienced engineering team on cutting-edge AI models and algorithms. The intern will collaborate on designing and testing inference solutions and optimizing models using Python and C++....Internship$100k - $150k
Tesla is seeking interns for the Performance Optimization team in Palo Alto, California, starting May 2026. The role involves optimizing AI models for Tesla’s AI-ASICs and collaborating with various teams. Ideal candidates will have strong skills in kernel optimization...Internship- Presto Phoenix, Inc. in Canada is seeking Fall 2026 AI Engineering Interns to help build the core intelligence behind our Voice AI platform, deployed across thousands of restaurant locations. This hands-on internship sits at the intersection of large language models, real...InternshipFull time
- ...a full-time internship in Palo Alto focusing on Performance Optimization. The role involves optimizing and implementing kernels for AI and Dojo ASICs, collaborating with cross-functional teams in compiling and hardware design. Ideal candidates should be pursuing a degree...InternshipFull time
- ...a high-growth Physical AI company transforming how... ...—captured through edge devices, mobile sensors, and... ...now hiring a Senior AI Engineer - Time-Series Signal Processing... ...and fluency with ML/DL frameworks (e.g., PyTorch,... ...and recommendations to internal and external stakeholders...Immediate start
$279.5k - $419.5k
AI Accelerator Software Distinguished Engineer - Framework Integration Ampere is a semiconductor design company for a new era, leading the future of computing with an innovative approach to CPU design focused on high-performance, energy efficient AI compute. As a pioneer...Local area- Mythic is building the future of AI computing with breakthrough... ...applications—whether in edge devices like drones, robotics, and... ...advanced signal processing, and is engineered to operate from -40 °C to +12... ...research and neural network frameworks. Here’s the background you...
- Junior AI Engineer, Agentic Orchestration Join to apply for the Junior AI Engineer, Agentic... ...to prototypes using modern agentic frameworks (LangChain, ReAct, AutoGPT) Collaborate... ...teams to ship intelligent features to HP’s devices Design and run experiments to fine-tune...Full time
- ...a senior C++/Linux expert to optimize and develop deep learning frameworks for AMD GPUs. You will work on GPU kernels, model performance, and multi-GPU/multi-node scales while collaborating with internal teams and open‑source maintainers. The ideal candidate combines deep...
- ...in California is seeking a motivated ML/AI intern to participate in the development of... ...opportunity to collaborate with experienced engineers, gaining hands-on experience in machine... ...with skills in programming and AI frameworks. Join us to shape the future of technology...Internship
$50 - $70 per hour
Tesla is seeking an intern for their AI Hardware Engineering team in Palo Alto, California. This position involves designing AI-powered applications that enhance EDA tools and collaborating with both software and hardware teams. Candidates should be pursuing a degree in...InternshipHourly pay- We're seeking an AI Data and Python Tools Engineer to develop and deploy intelligent tools that leverage big... ...React, and modern machine learning frameworks Design and streamline data... ...Basic UX/UI design sensibilities for internal tooling Experience with real‑time data...Full timeRelocation packageMonday to Friday
$206.5k - $258.1k
...to build and scale our AI solutions driving impact... ...teams. As a Staff AI Engineer, you will design, build... ...generation, evaluation frameworks, and scalable serving.... ...media/website, network/device, recruiting system usage... ...Personal Data with (i) internal personnel who have a need...Full timeContract workTemporary workPart timeLocal areaShift work$182k - $273k
Ampere Computing LLC. in Santa Clara seeks an AI Accelerator Software Principal Engineer to integrate and optimize AI frameworks for their deep learning accelerators. This role demands expertise in Python and C/C++, along with strong performance engineering skills. Successful...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to On-device AI Frameworks Engineer (Intern). Be the first to apply!

