Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

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

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Staff On-Device AI Frameworks Engineer in Palo Alto, CA vacancy
  • 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... 
    Suggested
    Work at office

    Argmax, Inc.

    Palo Alto, CA
    1 day ago
  • 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... 
    Suggested
    Internship
    Work at office
    3 days per week

    Argmax

    Palo Alto, CA
    1 day ago
  • 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... 
    Suggested
    Work at office
    Remote work
    3 days per week

    Argmax

    Palo Alto, CA
    1 day ago
  • 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

    Google Inc.

    Mountain View, CA
    2 days ago
  • 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

    Majestic Labs

    Los Altos, CA
    1 day ago
  • $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... 

    GEICO

    Palo Alto, CA
    1 day ago
  • $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 start
    Visa sponsorship
    Flexible hours

    Ford Motor Company

    Palo Alto, CA
    1 day ago
  • $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++... 

    Google Inc.

    Mountain View, CA
    2 days ago
  • $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 pay
    Full time
    Local area
    Remote work

    Rivian Automotive, Inc.

    Palo Alto, CA
    1 day ago
  • 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

    Jobright.ai

    Palo Alto, CA
    1 day ago
  • 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... 

    AMD

    Santa Clara, CA
    1 day ago
  • 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

    BrightAI Corporation

    Palo Alto, CA
    2 days ago
  • $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

    Ampere

    Santa Clara, CA
    1 day ago
  • 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... 

    Mythic

    Palo Alto, CA
    1 day ago
  • 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 time
    Relocation package
    Monday to Friday

    OSI Engineering

    Cupertino, CA
    1 day ago
  • $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

    Ampere Computing LLC.

    Santa Clara, CA
    1 day ago
  • $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 time
    Contract work
    Local area

    Rivian

    Palo Alto, CA
    1 day ago
  • $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 time
    Contract work
    Temporary work
    Part time
    Local area
    Shift work

    Rivian

    Palo Alto, CA
    1 day ago
  • $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... 

    Ampere Computing LLC.

    Santa Clara, CA
    1 day ago
  • $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 office
    Flexible hours

    Kodiak

    Mountain View, CA
    1 day ago
  • $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 office
    Work from home
    Relocation

    xAI

    Palo Alto, CA
    more than 2 months ago
  • 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

    Applied Intuition

    Sunnyvale, CA
    1 day ago
  •  ...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

    Unconventional AI

    Palo Alto, CA
    1 day ago
  • $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

    Intel

    Santa Clara, CA
    1 day ago
  •  ...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... 

    Intelliswift - An LTTS Company

    Mountain View, CA
    1 day ago
  • $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

    L&T Technology Services

    Mountain View, CA
    1 day ago
  • 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... 

    Tesla Motors, Inc.

    Palo Alto, CA
    1 day ago
  • $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)... 

    NVIDIA

    Santa Clara, CA
    1 day ago
  • $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 pay
    Full time
    Temporary work
    Flexible hours

    Tesla Motors, Inc.

    Palo Alto, CA
    1 day ago
  • $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

    Waymo

    Mountain View, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff On-Device AI Frameworks Engineer. Be the first to apply!