jax Jobs

1 - 20 of 231 Jobs

AI Engineer, Multimodal Pretraining, Autonomy & RoboticsJob Title - AI Engineer, Multimodal Pretraining, Autonomy & Robotics

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Scaling transformers, as well as more recent advances in Reinforcement Learning with Verifiable Rewards (RLVR), has created models with Ph-D level intelligence in a wide variety of subject areas - from Math to Social Sciences. Yet these models continue to struggle in real-world physical reasoning, often struggling to tell left from right. At Tesla AI, we want to develop Olympiad-level physical intelligence that will enable highly capable robots, both wheeled and legged. These models should be ab

AI Engineer, Reinforcement Learning & Distillation, Autonomy & RoboticsJob Title - AI Engineer, Reinforcement Learning & Distillation, Autonomy & Robotics

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Systems Engineer (Sys Integr, Ver & Validation)Job Title - Systems Engineer (Sys Integr, Ver & Validation)

Boeing CompanyCompany Name - Boeing Company

•

Richardson, Texas, USA

Full-time

Systems Engineer (Sys Integr, Ver & Validation) Company: The Boeing Company The Boeing Defense Space and Security (BDS) is looking for a Mid-Level Systems Engineer - Software Integration Lead Coordinator to join their team in Richardson, TX. This position will serve as an individual contributor for the P-8A Advanced Airborne Sensor (AAS) Training Systems Integration and Test team. The Integration and Test team provides software integration for the P-8A AAS Training Systems, which consists of

Lead Research EngineerJob Title - Lead Research Engineer

Aduril IndustriesCompany Name - Aduril Industries

•

Costa Mesa, California, USA

Full-time

Anduril Industries is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st century's most innovative companies to the defense industry, Anduril is changing how military systems are designed, built and sold. Anduril's family of systems is powered by Lattice OS, an AI-powered operating system that turns thousands of data streams into a realtime, 3D command and c

Lead Research EngineerJob Title - Lead Research Engineer

Aduril IndustriesCompany Name - Aduril Industries

•

Mountain View, California, USA

Full-time

Senior ML Compiler EngineerJob Title - Senior ML Compiler Engineer

General MotorsCompany Name - General Motors

•

Mountain View, California, USA

Full-time

Job Description Remote: This role is based remotely but if you live within a 50-mile radius of Mountain View, you are expected to report to that location three times a week, at minimum. Role: We are looking for a deep learning compiler engineer to build out our ML compiler for deploying machine learning models to a variety of ML hardware accelerators. You will develop and enhance GM's internal ML compiler for high performance, usability, and retargetability by leveraging open-source technology

Senior Staff Machine Learning Engineer - Driver Pricing & Marketplace OptimizationJob Title - Senior Staff Machine Learning Engineer - Driver Pricing & Marketplace Optimization

Uber CorporateCompany Name - Uber Corporate

•

New York, New York, USA

Full-time

About the Role We're seeking an exceptional Senior Staff ML Engineer to lead breakthrough ML innovation in Uber's Driver Pricing organization. This is a high-impact role where you'll architect and build next-generation ML systems that directly optimize marketplace efficiency and driver earnings for millions of drivers globally. You'll tackle some of the most complex problems in applied machine learning: real-time pricing optimization, supply-demand balancing, and driver behavior modeling at un

Lead Research EngineerJob Title - Lead Research Engineer

Aduril IndustriesCompany Name - Aduril Industries

•

Seattle, Washington, USA

Full-time

Principal Machine Learning Engineer (NLP - AI Runtime Security)Job Title - Principal Machine Learning Engineer (NLP - AI Runtime Security)

PaloAlto NetworksCompany Name - PaloAlto Networks

•

Santa Clara, California, USA

Full-time

Company Description Our Mission At Palo Alto Networks everything starts and ends with our mission: Being the cybersecurity partner of choice, protecting our digital way of life. Our vision is a world where each day is safer and more secure than the one before. We are a company built on the foundation of challenging and disrupting the way things are done, and we're looking for innovators who are as committed to shaping the future of cybersecurity as we are. Who We Are We take our mission of

Software Engineer, ML Inference Compiler & Deployment, GPU, CPUJob Title - Software Engineer, ML Inference Compiler & Deployment, GPU, CPU

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world for Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compiler to

Software Engineer, ML Inference Compiler & Deployment, AI FrameworksJob Title - Software Engineer, ML Inference Compiler & Deployment, AI Frameworks

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world, powering Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compi

Lead Research EngineerJob Title - Lead Research Engineer

Aduril IndustriesCompany Name - Aduril Industries

•

Boston, Massachusetts, USA

Full-time

Machine Learning Kernel Performance Engineer, DojoJob Title - Machine Learning Kernel Performance Engineer, Dojo

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a member of the Dojo Machine Learning software team, you will be responsible for developing and optimizing machine learning kernels to run on a massively parallel machine. The ideal candidate will have a strong background in kernel development and performance optimization for AI workloads, with a passion for delivering high-performance implementations, which are simple to use. Responsibilities Develop and validate datapath kernels on a massively parallel machineDesign and deliver implementat

Lead Research EngineerJob Title - Lead Research Engineer

Aduril IndustriesCompany Name - Aduril Industries

•

Washington, District of Columbia, USA

Full-time

Senior Machine Learning Engineer, Rider ExperienceJob Title - Senior Machine Learning Engineer, Rider Experience

Uber CorporateCompany Name - Uber Corporate

•

San Francisco, California, USA

Full-time

About the Role Rider Experience drives and enables the critical trip booking funnel within the Rides app that makes up almost all of the trip transactions and contributes tremendously to business growth. We actively explore algorithmic improvements to how we help millions of riders find and discover the right products every hour to move around the world, and power smart and intuitive experiences for them. Senior Softwareengineers at Uber are passionate and pragmatic technologists who are able

Software Engineer, ML Inference Compiler & Deployment, AI AcceleratorJob Title - Software Engineer, ML Inference Compiler & Deployment, AI Accelerator

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Research Support EngineerJob Title - Research Support Engineer

Booz Allen HamiltonCompany Name - Booz Allen Hamilton

•

Maryland, USA

Full-time

Job Number: R0222686 Research Support Engineer The Opportunity: As an experienced engineer, you know that machine learning is critical to understanding and processing massive datasets. Your ability to test and evaluate ML infrastructure and algorithms makes you an integral part of delivering a customer-focused solution. We need your technical knowledge and desire to problem-solve to support the government mission. Join us. The world can't wait. You Have: Experience with supporting and debuggin

Ray Inference EngineerJob Title - Ray Inference Engineer

VeearCompany Name - Veear

•

Remote

Contract

Position Summary: Designing, implementing, and maintaining distributed systems to build world-class ML platforms/products at scaleExperiment with, deploy, and manage LLMs in a production contextBenchmark and optimize inference deployments for different workloads, e.g. online vs. batch vs. streaming workloadsDiagnose, fix, improve, and automate complex issues across the entire stack to ensure maximum uptime and performanceDesign and extend services to improve functionality and reliability of the

Principal AI/ML Engineer - Onboard Embodied AIJob Title - Principal AI/ML Engineer - Onboard Embodied AI

General MotorsCompany Name - General Motors

•

Mountain View, California, USA

Full-time

Job Description Hybrid: This role is categorized as hybrid. This means the successful candidate is expected to report to the Mountain View Technical Center in the Bay Area three times per week, at minimum. Role: As a Technical Lead in Machine Learning within the Onboard Embodied AI organization, you will be a senior individual contributor driving cutting-edge end-to-end machine learning solutions directly impacting autonomous driving performance. Your role is pivotal in designing, architecting

AI/ML Engineer Sr. (mid-career)Job Title - AI/ML Engineer Sr. (mid-career)

Lockheed Martin CorporationCompany Name - Lockheed Martin Corporation

•

Stratford, Connecticut, USA

Full-time

Job Description The Lockheed Martin Artificial Intelligence Center (LAIC) is seeking a highly skilled leader to head the AI Consulting portfolio, a role that requires a unique blend of technical expertise in software engineering and AI/ML, as well as strong business acumen and communication skills. This is an exciting opportunity to lead the development and deployment of AI solutions that can drive business value and innovation across the Lockheed Martin enterprise. With a focus on collaboratio