JAX jobs in San Jose, CA

Staff Software Engineer, ML Infra, Autonomy

Palo Alto, California

•

Today

About Rivian Rivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate comfortably in areas that are unknown. Our backgrounds are diverse, but our team shares a love of the outdoors and a desire to protect it

Full-time

USD 206,500.00 - 258,100.00 per year

Machine Learning Lead Engineer, Machine Learning (ML) Required, Work From Home - G

Remote

•

Today

Machine Learning Lead Engineer, Machine Learning (ML) Required, Work From Home As Machine Learning Lead Engineer, you own the execution layer of our intelligence, turning research and model capabilities into reliable, scalable production systems. You will work across the model lifecycle: data, training, evaluation, inference, and deployment. This is a hands-on leadership role for someone who wants to operate at the intersection of research, systems, and product. Machine Learning (ML) experien

Easy Apply

Full-time

$160,000 - $190,000

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. MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT. VP of Research, Machine Learning (ML) Responsibilities:

Easy Apply

Full-time

$225,000 - $275,000

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home As Machine Learning Technical Lead, you own the execution layer of intelligence. You will translate research direction into reliable, scalable, production-grade ML systems. This role sits at the intersection of research, infrastructure, and product. You will be responsible for making models trainable, deployable, observable, and performant under real-world constraints. This position is 100% Remote. MUST BE

Easy Apply

Full-time

$190,000 - $225,000

Member of Technical Staff, Machine Learning, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Member of Technical Staff, Machine Learning, Artificial Intelligence (AI) Required, Work From Home As a Member of Technical Staff, Machine Learning, you will build core ML components. The Member of Technical Staff will work on real production systems from day one, learning how large-scale ML behaves outside of research settings. The Member of Technical Staff role is for engineers who want to develop strong systems judgment by shipping, debugging, and iterating on real-world ML. This position

Easy Apply

Full-time

$160,000 - $190,000

Principal Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Principal Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home As a Principal Machine Learning Engineer, you are a deep technical authority responsible for designing and evolving the most critical ML systems in the company. The Principal Machine Learning Engineer will operate across training, inference, evaluation, and infrastructure, solving the hardest architectural and performance problems. While Technical Leads may own execution at the team level, you set the te

Easy Apply

Full-time

$170,000 - $200,000

Machine Learning Platform Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Machine Learning Platform Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home As the Machine Learning Platform Engineer, you will build the infrastructure and systems that power Artificial Intelligence (AI) capabilities. You will design and operate the systems behind the Artificial Intelligence (AI) stack, from model training and evaluation to deployment, inference, observability, and continuous improvement. You will work closely with Artificial Intelligen

Easy Apply

Full-time

$140,000 - $180,000

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home As an LLM Application Engineer, you will build the intelligence layer that powers the AI experiences. You will work at the intersection of Large Language Models (LLMs), software engineering, and product designing agent workflows, improving model behavior, and turning Artificial Intelligence (AI) capabilities into reliable user experiences. You will own problems end-to-end, from understanding user needs, designing A

Easy Apply

Full-time

$130,000 - $175,000

Senior/Staff Engineer, Machine Learning - Online Mapping

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 - 291,150.00 per year

Staff Engineer, ML Profiling Tools

San Jose, California

•

Today

Please Note: To provide the best candidate experience amidst our high application volumes, each candidate is limited to 10 applications across all open jobs within a 6-month period. Advancing the World's Technology Together Our technology solutions power the tools you use every day--including smartphones, electric vehicles, hyperscale data centers, IoT devices, and so much more. Here, you'll have an opportunity to be part of a global leader whose innovative designs are pushing the boundaries o

Full-time

USD 163,000.00 - 253,000.00 per year

Principal Engineer, Efficient GenAI

San Jose, California

•

Today

ADVANCE YOUR CAREER. ADVANCE THE WORLD. At AMD, we believe technology has the power to solve the world's most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future. Whether you're designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger - technology that moves the world forward. Joi

Full-time

USD 210,000.00 per year

AI Researcher

Santa Clara, California

•

Today

Grow with us This role is an in-office position with the expectation to work onsite in the Ericsson office located in Santa Clara, CA and relocation is not offered for this position. About this opportunity Ericsson Research is growing its AI Silicon Valley team, working at the intersection of artificial intelligence and wireless technologies. The team advances technology leadership by applying AI and machine learning to the next generation of wireless communication systems. This role focuses

Full-time

USD 146,500.00 - 192,150.00 per year

Senior GPU Performance Engineer - AI Workloads

Santa Clara, California

•

Today

Full-time

USD 178,500.00 per year

Staff Software Engineer, Machine Learning (Consumer Revenue)

Remote or San Francisco, California

•

Today

Discord has a highly engaged community of millions of daily active users who use the platform for many different reasons, but there's one thing that nearly everyone does: play video games. Discord plays a uniquely important role in the future of gaming, and we are focused on making it easier and more fun for people to hang out before, during, and after playing games. We're looking to hire a Staff Machine Learning Engineer on our Consumer Revenue ML team. This team applies Machine Learning acros

Full-time

USD 272,000.00 - 340,000.00 per year

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 - 200,000.00 per year

2027 Summer Intern - AI/ML Engineer, Autonomous Vehicle: Simulation

Sunnyvale, California

•

Today

Job Description About the role: As a Simulation AI/ML Engineer intern, you'll partner with senior engineers to deliver key components of the evaluation strategy for GM's autonomous vehicle programs. In this role, you will scale the testing and evaluation of our self-driving stack by applying AI/ML techniques to software solutions. You will design, perform experiments, and deploy these solutions as production-ready code. The work you do will be directly applied to the iteration and validation of

Full-time

USD 8,100.00 - 9,600.00 per month

Distributed Systems Software Engineer, Collectives

San Jose, California

•

Today

Full-time

USD 150,500.00 per year

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

San Jose, California

•

Today

Responsibilities Team Introduction: 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 us for an internship. PhD internships at Our Company provide students with the opportunity to actively contribute to our products and research, as well as to the organization's future plans and emerging technologies. Our dynamic internship experienc

Full-time

USD 60.00 - 60.00 per hour

ML Systems Research Engineer, RL / Inference / Agent Systems

Santa Clara, California

•

Today

Full-time

USD 178,500.00 per year

Senior AI/ML Solutions Engineer

Santa Clara, California

•

Today

Full-time

USD 175,700.00 per year

JAX jobs in San Jose, CA, USA