jax Jobs

101 - 120 of 174 Jobs

Principal Solutions Engineer, Infrastructure (SLURM & AI Focus)Job Title - Principal Solutions Engineer, Infrastructure (SLURM & AI Focus)

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

Santa Clara, California, USA

Full-time

WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellenc

AIML - Machine Learning, Data and ML InnovationJob Title - AIML - Machine Learning, Data and ML Innovation

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Are you excited to tackle some of the most ambitious technical challenges in Apple Intelligence? Be involved in collaborating closely with our machine learning researchers, engineers, and data scientists? Together, you will orchestrate groundbreaking research initiatives and develop transformative products designed to build a significant impact for billions of users worldwide! As part of Apple's AI and Machine Learning org, we encourage and create groundbreaking technology for large-scale ML sys

On-device ML Infrastructure Engineer (Core ML Framework)Job Title - On-device ML Infrastructure Engineer (Core ML Framework)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The On-Device Machine Learning team at Apple is responsible for enabling the Research to Production lifecycle of cutting edge machine learning models that power magical user experiences on Apple's hardware and software platforms. Apple is the best place to do on-device machine learning, and this team sits at the heart of that discipline, interfacing with research, SW engineering, HW engineering, and products. The team builds critical infrastructure that begins with onboarding the latest machine

Senior/Staff Software Engineer, ML Infrastructure, OptimizationJob Title - Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro exists to better everyday life through robotics. Founded in 2016, Nuro has spent eight years developing autonomous driving (AD) technology and commercializing AD applications. The Nuro Driver is our world-class autonomous driving system that combines AD hardware with our generalized AI-first self-driving software. Built to learn and improve through data, the Nuro Driver is one of the few driverless autonomous technologies on public roads today. Nuro has raised over $2B in capi

AIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform & TechnologyJob Title - AIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform & Technology

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Apple is where individual imaginations gather together, committing to the values that lead to great work. Every new product we build, service we create, or Apple Store experience we deliver is the result of us making each other's ideas stronger. That happens because every one of us shares a belief that we can make something wonderful and share it with the world, changing lives for the better. It's the diversity of our people and their thinking that inspires the innovation that runs through every

On-device ML Infrastructure Engineer (ML User Experience APIs)Job Title - On-device ML Infrastructure Engineer (ML User Experience APIs)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Sr. Computer Vision & Machine Learning Engineer, Creativity AppsJob Title - Sr. Computer Vision & Machine Learning Engineer, Creativity Apps

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

At Apple, new ideas have a way of becoming phenomenal products, services, and customer experiences very quickly! The Creativity Apps team needs your help shaping the next generation of creative editing tools by working on pioneering technologies to surprise and delight creative pros and enthusiasts alike. As Senior Computer Vision & Machine Learning Engineer, you will be working alongside our world-class creatives, designers, and engineers to help innovate in the creative space in ways that only

Director Software Development - AI InferenceJob Title - Director Software Development - AI Inference

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

Bellevue, Washington, USA

Full-time

Staff Engineer, AI/ML Software CompilerJob Title - Staff Engineer, AI/ML Software Compiler

Samsung Electronics AmericaCompany Name - Samsung Electronics America

•

San Jose, California, USA

Full-time

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

ML Engineer - Generative AI, Siri Agent ModelingJob Title - ML Engineer - Generative AI, Siri Agent Modeling

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The Siri team is looking for passionate Machine Learning Engineers to join us in developing and shipping state-of-the-art generative AI technology to advance Siri and Apple Intelligence for Apple's customers. Siri is being elevated by the huge opportunities that AI brings. The team is responsible for training on-device & cloud models, evaluating various approaches, pushing the envelope with the latest generative AI research developments, and ultimately delivering product critical models that pow

Senior ML Engineer - Generative AI, Siri Agent ModelingJob Title - Senior ML Engineer - Generative AI, Siri Agent Modeling

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Senior High-Performance LLM Training EngineerJob Title - Senior High-Performance LLM Training Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for a Senior High-Performance LLM Training Engineer! NVIDIA is seeking experienced engineers specializing in performance analysis and optimization to improve the efficiency of LLM training workloads, which are shaping the world's most advanced computing systems. This position focuses on optimizing NVIDIA's high-performance LLM software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while also helping shape hardware roadmaps for th

Sr. ML Infrastructure Engineer, Apple Data PlatformJob Title - Sr. ML Infrastructure Engineer, Apple Data Platform

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The Apple Data Platform (ADP) group builds the data platform that enables the next generation of intelligent experiences on all Apple products and services. ADP empowers Apple engineers to deliver ML-driven products and innovations rapidly and at scale. We are looking for an experienced engineer who can bring their passion for machine learning, infrastructure, big data, and distributed systems to build world class data+ML platform/products at scale. You will work with many cross functional teams

AIML - Sr. ML Engineer, NL Response Generation - Answers, Knowledge, & Intelligence (AKI)Job Title - AIML - Sr. ML Engineer, NL Response Generation - Answers, Knowledge, & Intelligence (AKI)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The Siri and Information Intelligence team is revolutionizing the way hundreds of millions of people access information on their devices, all while keeping user privacy at the forefront. As an Applied ML team, we're pushing the boundaries of Apple Intelligence, result ranking, and innovative search technologies, all while running a low latency production service. Our work fuels intuitive information experiences across some of Apple's most iconic products, including Siri, Spotlight, Safari, Messa

Machine Learning Research Engineer, Generative AIJob Title - Machine Learning Research Engineer, Generative AI

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Sr Machine Learning Engineer, ML Systems EngineeringJob Title - Sr Machine Learning Engineer, ML Systems Engineering

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Apple's products combine the best hardware and incredible software to deliver magical experiences to our customers. The Proactive Intelligence team builds features that anticipate customer's needs and create personalized experiences by adapting to user behaviors with machine learning running locally on users' devices. We are seeking exceptional machine learning engineers with a strong software engineering background to join our team where we're making Apple Intelligence more capable and personal

Principal Machine Rendering EngineerJob Title - Principal Machine Rendering Engineer

Epic GamesCompany Name - Epic Games

•

Cary, North Carolina, USA

Full-time

WHAT MAKES US EPIC? At the core of Epic's success are talented, passionate people. Epic prides itself on creating a collaborative, welcoming, and creative environment. Whether it's building award-winning games or crafting engine technology that enables others to make visually stunning interactive experiences, we're always innovating. Being Epic means being a part of a team that continually strives to do right by our community and users. We're constantly innovating to raise the bar of engine an

Senior DGX AI Cloud Performance Analysis Tools EngineerJob Title - Senior DGX AI Cloud Performance Analysis Tools Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

Joining NVIDIA's DGX Cloud AI Efficiency Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of AI workloads, as well as developing scalable AI and Data infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the necessary resources and scale to foster innovation. We are seeking excellent Software Engineers to design and develop

Staff Software Development Engineer - AI/ML frameworks, C/C++/PythonJob Title - Staff Software Development Engineer - AI/ML frameworks, C/C++/Python

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

San Jose, California, USA

Full-time

Senior Machine Learning Engineer, NLPJob Title - Senior Machine Learning Engineer, NLP

Cisco Systems, Inc.Company Name - Cisco Systems, Inc.

•

No location provided

Full-time

The application window is expected to close on: June 22, 2025. Meet the Team Cisco Webex is the world's #1 business collaboration and customer engagement platform. It is used across all industries in the world being used to support governments, business, education, and more, including use by 85% of Fortune 500 business. The workplace of the future will be powered by AI, and we, Collaboration AI, is the team responsible for building AI/ML intelligent experiences within Cisco's Webex portfolio.