Staff On-Device AI Frameworks Engineer
Argmax
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 into production-ready solutions, while also supporting enterprise customers. The ideal candidate has 3+ years of experience in SDK or Frameworks development, fluent in Swift, and familiarity with AI technologies. Argmax offers competitive equity, healthcare, and flexible work arrangements. #J-18808-Ljbffr Argmax
- 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
- 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...SuggestedInternshipWork at office3 days per week
- 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
- 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
- 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...Suggested
$110k - $260k
GEICO is seeking a Staff Security Engineer for their Red Team, focused on enhancing detection and response through AI-driven adversary operations. The ideal candidate will have extensive... ...in a relevant field and expertise in AI frameworks are required, with a salary range of $11...$167.2k - $316.6k
...development of system‑level components and libraries within the Android framework. Build HAL, AIDL interfaces and Android services. Work... ...Bachelor’s or Master’s degree in Computer Science, Software Engineering, or equivalent combination of relevant education and...Immediate startVisa sponsorshipFlexible 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++...$186k - $232.5k
...Rivian supports developing and deploying Rivian’s Smart Diagnostics AI system that drives in-vehicle and remote diagnostics using... ...quality software and backend services, enhancing the utility of GenAI frameworks, and exploring the bleeding edge of this new technology....Hourly payFull timeLocal areaRemote work- 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
- AMD is seeking 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...
- Senior AI Engineer, Time-Series Signal Processing Senior AI Engineer, Time-Series Signal Processing... ...-world events—captured through edge devices, mobile sensors, and large-scale cloud... ...in Python and fluency with ML/DL frameworks (e.g., PyTorch, TensorFlow, Keras). Experience...Immediate start
$279.5k - $419.5k
...to CPU design focused on high-performance, energy efficient AI compute. Description As a pioneer in the new frontier of energy... ...About The Role As an AI Accelerator Software Distinguished Engineer - Framework Integration, you will lead end-to-end technical strategy and...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...
- Responsibilities Develop and deploy AI-powered full-stack applications using Python, React, and modern machine learning frameworks Design and streamline data pipelines, train... ...development lifecycle Software Engineering & AI/ML Data, Tools Development 3+ years...Full timeRelocation packageMonday to Friday
$182k - $273k
AI Accelerator Software Principal Engineer - Framework Integration Ampere is a semiconductor design company focused on high-performance, energy efficient AI compute. As a pioneer in energy efficient high‑performance computing, Ampere is part of the Softbank Group driving...Local area$265k - $331.3k
...scale our Enterprise Applied AI solutions driving impact across internal teams. As a Staff AI Engineer (Staff or Senior Staff), you... ...augmented generation, evaluation frameworks, and scalable serving.... ...social media/website, network/device, recruiting system usage/interaction...Full timeContract workLocal area$206.5k - $258.1k
...contributor to build and scale our AI solutions driving impact... ...Technology teams. As a Staff AI Engineer, you will design, build, and... ...augmented generation, evaluation frameworks, and scalable serving. Build... ...media/website, network/device, recruiting system usage/interaction...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...$190k - $260k
...an artificial intelligence (AI) powered technology stack purpose... .... We are looking for engineers who make model training fast:... ...models that don't fit on a single device Maximize utilization of modern... ...with distributed training frameworks and techniques (PyTorch DDP/FSDP...Work at officeFlexible hours$180k
...About xAI xAI’s mission is to create AI systems that can accurately understand the... ...small, highly motivated, and focused on engineering excellence. This organization is for... ...at xAI builds an end to end RL training framework to enable pretrain scale RL. About the...Work at officeWork from homeRelocation- Applied Intuition in Sunnyvale, California is seeking a skilled engineer to work on on-device intelligence for the Android Automotive platform. The role includes deploying ML systems, optimizing models under real-world constraints, and interfacing with vehicle signals....Flexible hours
- ...Unconventional Since 2022, AI has entered the mainstream... ...networks directly to the device physics rather than relying... ...As a Member of Technical Staff, Silicon Validation Engineer, you will be a foundational... ...Python-based automation frameworks to accelerate characterization...Work at office
$170.5k - $315.49k
Intel in Santa Clara is seeking a candidate to design and implement an agent framework supporting hybrid local and cloud AI models. The role involves building the agent loop, managing context, tool invocation, and ensuring security and observability within the harness....Local area- ...firm in Mountain View, CA is seeking a highly skilled Device-Side Test Automation Engineer to join their hardware validation team. The successful... ...responsible for developing and maintaining automation frameworks for critical device subsystems like Bluetooth and Wi-Fi...
$80k - $110k
...automated tests to ensure software quality. The ideal candidate will have strong proficiency in Python and experience with mobile device stacks. The position offers a competitive salary range of $80,000 to $110,000 annually based on skills and experience, and is full-...Full time- Tesla Motors, Inc. is seeking an expert to join its Deployment Infrastructure team, enhancing system performance crucial for AI model evaluations. The role demands hands-on programming in Python or Go, a strong foundation in Linux and cloud infrastructure, and innovative...
$184k - $287.5k
...the most significant capability gains in AI today. It is the process that teaches a... ...can do. NVIDIA is building an RL Frameworks engineering team to develop the open-source tools and... ...routine operations (custom operators, GPU device plugins, or scheduling contributions)...$140k - $300k
...Enhance fault tolerance and improve/design new redundancy‑aware frameworks in mature autonomous systems Collaborate with other teams to... ...Bachelor's Degree in Computer Science, Physics, Computer Engineering, Electrical Engineering or proof of exceptional skills in related...Hourly payFull timeTemporary workFlexible hours$204k - $259k
...of billions in simulation across 15+ U.S. states. The Waymo ML Frameworks & Efficiency team partners with Research and Production teams... ...problems unique to ML for autonomous driving. We are looking for engineers with ML system expertise to help us train and improve pre-...Full time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff On-Device AI Frameworks Engineer. Be the first to apply!
