JAX jobs in San Jose, CA

Remote or Santa Clara, California

•

Today

NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code. Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX. To achieve this and provide an excellent user experience, you have strong programming skills and excellent

Full-time

USD 184,000.00 per year

AIML - Machine Learning Researcher, Foundation Models

Cupertino, California

•

Today

We are a group of engineers and researchers responsible for building foundation models at Apple. We build infrastructure, datasets, and models with fundamental general capabilities such as understanding and generation of text, images, speech, videos, and other modalities and apply these models to Apple products. We are looking for researchers who are passionate about developing algorithms, techniques, and systems that push the frontier of deep learning and delight millions of users with Apple pr

Full-time

USD 147,400.00 - 272,100.00 per year

AIML - Senior Software Engineer, Foundation Model Training - ML Platform Technologies

Santa Clara, California

•

Today

We are a group of engineers to support training foundation models at Apple.! We build infrastructure to support training foundation models with general capabilities such as understanding and generation of text, images, speech, videos, and other modalities and apply these models to Apple products. We are looking for engineers who are passionate about building systems that push the frontier of deep learning in terms of scaling, efficiency, and flexibility and delight millions of users in Apple pro

Full-time

USD 181,100.00 - 318,400.00 per year

AIML - Staff Software Engineer, Foundation Model Training - ML Platform Technologies

Santa Clara, California

•

Today

Full-time

USD 181,100.00 - 318,400.00 per year

AIML - Staff Machine Learning Engineer - ML Efficiency, ML Platform & Technology

Santa Clara, California

•

Today

Do you want to shape the platform that enables the next generation of intelligent experiences on Apple products & services? In Apple's Machine Learning Platform Technology & Infra team we have built the platform that Apple uses for developing machine learning, artificial intelligence, and computer vision applications. As a team, we have a variety of technical backgrounds, from machine learning PhDs to builders of large-scale production systems. Specifically in this role you will be working on op

Full-time

USD 181,100.00 - 318,400.00 per year

AI/ML - Machine Learning Research Engineer, Machine Translation

Cupertino, California

•

Today

Apple Machine Translation team is looking for exceptional researchers/scientists to develop the next generation of core MT technology to allow our users to communicate across language barriers. Passionate about Natural Language Processing, Language Modeling and Machine Translation? Join us to bring the most advanced Machine Translation solution across the Apple ecosystem. Description Machine Translation is a critical technology that enables connecting people across language barriers. The Apple M

Full-time

USD 212,000.00 - 318,400.00 per year

AIML - Machine Learning Engineer, Answers, Knowledge & Information (AKI)

Santa Clara, California

•

Today

Do you want to make Siri and Apple products smarter for our users? The Answers, Knowledge & Information team is redefining how hundreds of millions of people use their devices to get information. We are an Applied ML team pushing the limits on realtime augmented information retrieval and generation, information safety and search technologies, while also responsible for a few user facing production services. We are part of a wider effort to power information across a variety of Apple products - i

Full-time

USD 147,400.00 - 272,100.00 per year

Technical Lead, ML Training Infrastructure

Mountain View, California

•

Today

Who We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver , to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms

Full-time

USD 222,775.00 - 333,925.00 per year

AIML - Machine Learning Engineer- Health AI

Santa Clara, California

•

Today

The Health AI team is at the forefront of machine learning and health science at Apple. We are a close-knit team of accomplished, deeply technical research scientists, software engineers and machine learning engineers, passionate about delivering innovative technologies that impact millions of users. We are looking for a senior engineer passionate about solving real-world problems in the health domain that make a difference in our customers' lives. Description In this role, you will use your ski

Full-time

USD 212,000.00 - 386,300.00 per year

Machine Learning Engineer, Siri Speech

Cupertino, California

•

Today

Are you excited about Generative AI and Large Language Models? Are you interested in working on cutting-edge generative modeling technologies-plus the high-quality data that fuels them-to enrich billions of people? Join the Siri Speech team at Apple! We are looking for an exceptional senior ML engineer experienced in collecting and preparing large-scale speech data, training/adapting models, and deploying them in production. You'll advance the latest ML techniques, shape the datasets that power

Full-time

USD 147,400.00 - 272,100.00 per year

Technical Lead Manager, ML Training Infrastructure

Mountain View, California

•

Today

Full-time

USD 222,775.00 - 333,925.00 per year

On-device ML Infrastructure Engineer (Core ML Framework)

Cupertino, California

•

Today

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 l

Full-time

USD 126,800.00 - 220,900.00 per year

Senior Engineer - Deep Learning Compiler Verification and Infrastructure

Santa Clara, California

•

Today

We are building the next generation of compiler technologies to accelerate deep learning workloads. We are looking for an engineer to implement compiler verification software & related infrastructure in the AI space. You will be solving critical problems working alongside a diverse set of minds in GPU computing and systems software, doing what you enjoy. If this sounds like a fun challenge, we want to hear from you! What you'll be doing: In this role you will work closely with compiler develop

Full-time

USD 136,000.00 per year

AIML - Machine Learning Engineer, Foundation Models

Cupertino, California

•

Today

Apple is revolutionizing artificial intelligence by developing sophisticated foundation models that power intelligent features across our product ecosystem. We're seeking skilled Machine Learning Engineers to transform cutting-edge research into scalable, production-ready AI solutions. We are looking for engineers who are passionate about building systems that push the frontier of deep learning in terms of scaling, efficiency, and flexibility and delight millions of users in Apple products. Desc

Full-time

USD 181,100.00 - 318,400.00 per year

Principal Machine Learning Engineer (NLP - AI Runtime Security)

Santa Clara, California

•

Today

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

Full-time

USD 151,000.00 - 246,500.00 per year

Senior Deep Learning Software Engineer, FlashInfer

Santa Clara, California

•

Today

We are now looking for a Senior Deep Learning Software Engineer, FlashInfer. NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's n

Full-time

USD 184,000.00 per year

Applied Research Engineer - Multimodal LLMs for Human Interaction

Sunnyvale, California

•

Today

Are you excited about the amazing potential of foundation models, LLMs, and multimodal LLMs? We are looking for individuals who thrive on collaboration and have a desire to push the boundaries of what is possible today! The Video Computer Vision org is a centralized applied research and engineering organization responsible for developing real-time on-device Computer Vision and Machine Perception technologies across Apple products. We balance research and product to deliver Apple quality, pioneer

Full-time

USD 147,400.00 - 272,100.00 per year

Machine Learning Engineer (Search)

Cupertino, California

•

Today

Apple Maps and the thousands of applications it empowers are being used by millions every single day! As a fundamental tool for human activity, Maps technology is evolving and new techniques are emerging. We are looking for a Machine Learning Engineer to join and play a big part in the next revolution of Maps; to enable users to find more things in innovative ways. On our team, you will have plenty of opportunities to build groundbreaking technologies using Machine Learning and Generative AI at

Full-time

USD 147,400.00 - 272,100.00 per year

Software Development Manager, AWS Neuron Frameworks

Cupertino, California

•

Today

We're seeking a Software Development Manager to lead our Frameworks team within AWS Neuron, the software stack powering AWS Inferentia and Trainium machine learning accelerators. This role combines technical leadership, team management, and strategic open-source collaboration to shape the future of machine learning acceleration at AWS. As the Software Development Manager, you will lead and mentor a high-performing team of software engineers while driving the development and maintenance of criti

Full-time

USD 166,400.00 per year

AIML - Staff Machine Learning Engineer, Answers Knowledge and Information

Cupertino, California

•

Today

The AIML Information Intelligence team is creating groundbreaking technology for artificial intelligence, machine learning and natural language processing! The features we create are redefining how hundreds of millions of people use their computers and mobile devices to search and find what they are looking for. Our universal search engine powers search features across a variety of Apple products, including Siri, Spotlight, Safari, Messages and Lookup. We also develop state-of-the-art generative

Full-time

USD 212,000.00 - 386,300.00 per year

Filter Results

Job post features

Posted date

Work settings

Employment type

Distance

Employer type

Work authorization

JAX jobs in San Jose, CA, USA