JAX jobs in San Jose, CA

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Mountain View, California

•

Today

Who We Are Nuro believes self-driving vehicles are the most immediate and profound opportunity for AI to drive positive change in the physical world. Safer streets, more time for what matters, and easier access to the world around us, that's why we're building a universal autonomy platform: self-driving for all roads and all rides. Founded in 2016, Nuro is a physical AI company developing Level 4 autonomous driving technology for a wide range of vehicles, use cases, and markets. Powered by the

Full-time

USD 193,930.00 - 352,290.00 per year

Senior Staff Research Engineer - Reinforcement Learning for AI Agents

Santa Clara, California

•

Today

XPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take-off and landing (eVTOL) aircraft, and robotics. With a strong focus on intelligent mobility, XPENG is dedicated to reshaping the future of transportation through cutting-edge R&D in AI, machine learning, and smart connectivity. We are looking for exceptional Research Engineers / Scient

Full-time

USD 244,140.00 - 413,160.00 per year

AI Research Scientist - Infrastructure Engineer, Reinforcement Learning

Santa Clara, California

•

Today

WHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you'll discover the real differentiator is our culture. We push the limits of innovation to solve the wo

Full-time

USD 178,500.00 per year

Sr. Software Development Engineer

San Jose, California

•

Today

ADVANCE YOUR CAREER. ADVANCE THE WORLD. At AMD, we believe technology can change lives for the better. It can heal us, entertain us, and make us more connected, productive, and understanding of the world around us. And we're looking for talent who feel the same: people who want to leave the planet better than they found it, those who don't shy away from humanity's challenges but are determined to help solve them. AMD is powering the next generation of supercomputing, high-performance computing

Full-time

USD 127,400.00 per year

LLM Post-training Engineer Graduate (Research & Product) - 2027 Start

San Jose, California

•

Today

Responsibilities About the team The Multimedia AI team at TikTok focuses on building, researching, and applying Large Language Models (LLMs) to power our global products. We believe the most impactful problems in AI arise at the intersection of research and real-world deployment-and post-training is where that intersection is sharpest. We are looking for talented individuals to join our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock l

Full-time

USD 128,000.00 - 316,800.00 per year

Senior / Staff AI Research Scientist, Foundation Models

Milpitas, California

•

Today

Why RoboForce RoboForce is an AI robotics company developing Physical AI-powered Robo-Labor for dull, dirty, and dangerous work. The company's robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability. We are looking for a Senior / Staff AI Research Scientist, Foundation Models to advance robotic embodied intelligence. In this role, you will develop algorithms that enable robots to understand their environment, interpret and execute task

Full-time

USD 200,000.00 - 360,000.00 per year

Software Engineer, ML Infrastructure, Optimization

Mountain View, California

•

Today

Full-time

USD 160,360.00 - 240,540.00 per year

Sr. Staff Software Development Engineer - Collectives and Network optimization

San Jose, California

•

Today

Full-time

USD 178,500.00 per year

LLM Post-Training Engineer Graduate (Research & Product) - 2027 Start

San Jose, California

•

Today

Full-time

USD 128,000.00 - 316,800.00 per year

AI Robotics Researcher Intern (Dexterous Manipulation)

San Jose, California

•

Today

JOB DESCRIPTION About NIO NIO is a pioneer and a leading company in the premium smart electric vehicle market. Founded in November 2014, NIO's mission is to shape a joyful lifestyle. NIO aims to build a community starting with smart electric vehicles to share joy and grow together with users. NIO designs, develops, jointly manufactures and sells premium smart electric vehicles, driving innovations in next-generation technologies in autonomous driving, digital technologies, electric powertrains

Full-time

USD 38.00 - 46.00 per hour

Senior Machine Learning Engineer, Perception

Santa Clara, California

•

Today

PlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. Headquartered in Silicon Valley with operations in the United States and Europe, Plus was named by Fast Company as one of the World's Most Innovative Companies. Partners including TRATON GROUP's Scania, MAN, and International brands, Hyundai Motor Company, Iveco Group, Bosch, and DSV are working with Plus to accelerate the deployment of next-generation autonomous trucks. If you're rea

Full-time

USD 150,000.00 - 250,000.00 per year

ML Systems Research Engineer, RL / Inference / Agent Systems

Santa Clara, California

•

Today

Full-time

USD 178,500.00 per year

Software Engineer- GPU/AI/ML

Santa Clara, California

•

Today

Full-time

USD 178,500.00 per year

Staff Software Development Engineer: GPU, Computer Vision, AI/ML Ops

Santa Clara, California

•

Today

Full-time

USD 150,500.00 per year

Fellow Software Engineer - AI Performance & Reliability

San Jose, California

•

Today

Full-time

USD 234,500.00 per year

Forward Deployed AI Research Scientist

Santa Clara, California

•

Today

Full-time

USD 178,500.00 per year

Machine Learning Engineer Graduate (E-Commerce Recommendation Mall) - 2027 Start

San Jose, California

•

Today

Responsibilities Our E-commerce Recommendation Team is responsible for building up and scaling our recommendation system to provide the best shopping experience for our TikTok users. We are looking for talented individuals to join our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth. Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation d

Full-time

USD 128,000.00 - 316,800.00 per year

Senior/Staff Machine Learning Engineer, Sensor Simulation

Mountain View, California

•

Today

Full-time

USD 193,930.00 - 291,150.00 per year

Senior / Staff AI Research Engineer, Data Infrastructure

Milpitas, California

•

Today

Why RoboForce RoboForce is an AI robotics company developing Physical AI-powered Robo-Labor for dull, dirty, and dangerous work. The company's robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability. We are looking for a Senior / Staff AI Research Engineer, Data Infrastructure to build the data and learning engine behind RoboForce's Physical AI stack. In this role, you will own the full pipeline - from raw teleoperation and UMI device

Full-time

USD 185,000.00 - 325,000.00 per year

VP of Research, Machine Learning (ML), Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

VP of Research, Machine Learning (ML), Artificial Intelligence (AI) Required, Work From Home The VP of Research, Machine Learning (ML) will own the research and intelligence direction of this system. The VP of Research, Machine Learning (ML) will define how Artificial Intelligence (AI) reasons, evaluates, and improves in a product used with high frequency. This position is 100% Remote. VP of Research, Machine Learning (ML) Responsibilities: - Set and evolve the research direction for the core i

Easy Apply

Full-time

$225,000 - $275,000

JAX jobs in San Jose, CA, USA