Jax jobs in san jose, ca

Neuron Runtime Software Development Engineer , Neuron Runtime

Cupertino, California

•

Today

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the servers that use them.As the Software Development Engineer for the Neuron Runtime Team, you will be responsible for working alongside a team of engineers to develop and maintain high-performance runtime libraries and drivers for machine learning applications and AI accelerators. You will work on design, development, and deployment of Neuron Runtime and other Neuron comp

Full-time

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

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

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

San Jose, California

•

Today

Full-time

USD 128,000.00 - 316,800.00 per year

LLM Post-training Engineer Intern (Research & Product) - 2027 Summer

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 us for an internship. Our internship program offers students hands-on experience, industry exposure, and opportun

Full-time

USD 45.00 - 45.00 per hour

Senior/Staff Machine Learning Engineer, Sensor Simulation

Mountain View, California

•

Today

Full-time

USD 193,930.00 - 291,150.00 per year

Staff/Sr. ML Infrastructure Engineer, Foundation Model Compute Infra

Cupertino, California

•

Today

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

Full-time

AIML - ML Prototyping Engineer, Machine Learning Research

Cupertino, California

•

Today

The Machine Learning Research Prototyping team sits at the intersection of cutting-edge research and real-world impact. We serve as the bridge between groundbreaking AI and ML discoveries and the innovative products that define tomorrow's user experiences. Our nimble, collaborative team transforms theoretical breakthroughs into tangible prototypes that help Apple understand where AI could go next, and help the broader ML community build on Apple research. We're a small team of engineers who ble

Full-time

Perception Algorithm Engineer

Cupertino, California

•

Today

Full-time

ML Framework (MetalLM) Engineer

Cupertino, California

•

Today

Apple's ML Frameworks (MetalLM) team in GPU, Graphics and Machine Learning works on enabling Apple Intelligence through high-performance, distributed inference of GenAI applications (such as LLMs) on Private Cloud Compute. You will get to work on custom-built server hardware that brings the power and security of Apple silicon to the data center. Team also works on GPU acceleration of ML Training frameworks such as PyTorch and JAX using Metal runtime and device backend. We are looking for engine

Full-time

Senior/Staff Machine Learning Engineer, Search & Knowledge Platforms

Santa Clara, California

•

Today

Are you passionate about search technologies and building knowledge experiences? The Answers, Knowledge, and Information team is at the forefront of revolutionizing how hundreds of millions of people use their devices to obtain information. We are a world-class team of machine learning engineers who collaborate closely with product, data science, and infrastructure teams to power and enhance features across Apple products, including Siri, Spotlight, Safari, Messages, and more. Our team operates

Full-time

AI/ML Performance Analysis and Validation Engineer

Cupertino, California

•

Today

Full-time

Software Engineer, Search and RAG Applications

Santa Clara, California

•

Today

At Apple, individual imaginations come together to build innovative products, services, and experiences that enrich lives worldwide. By encouraging a culture of collaboration and diversity, we strengthen each other's ideas, unlocking innovation that runs through everything we do. The AI, Search & Knowledge Platforms team is looking for a Software Engineer to help develop a best-in-class service for real-time search to power the capabilities of Siri and other Apple search products. These services

Full-time

Machine Learning Engineer

Sunnyvale, California

•

Today

Imagine what you could do here. At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish. Do you want to make Siri and Apple products more intelligent for our users? The Machine Learning Platform Technology team is building groundbreaking technology for search, natural language processing, artificial intelligence and machine learning. Our infrastructure

Full-time

Sr. Machine Learning Engineer, Foundation Models Inference - Cloud OS & Inference

Santa Clara, California

•

Today

We are the Foundation Model Inference team within Cloud OS and AI Inference organization. We are on a mission to build the most highly performant, secure and private inference stack that powers Siri AI, Apple Intelligence and Apps that are powered with the largest foundation models. Our systems serve billions of queries daily across Siri AI, Apple Intelligence, Apple Search, Apple Music, Apple TV, App Store, iMessage, Photos, Camera, Spotlight & Safari, at remarkably low latency with every ounc

Full-time

AIML - ML Engineer, Responsible AI and Safety

Cupertino, California

•

Today

Join a team at the forefront of defending Apple's ecosystem. We build the large-scale machine learning systems that protect millions of users from emerging threats and ensure the integrity of our products. We are looking for an experienced Applied ML Engineer who has a proven track record of shipping production models. The ideal candidate is passionate about tackling complex safety and security challenges using state-of-the-art techniques. In this role, you will design, build, and deploy the cri

Full-time

ML Engineer, Apple Foundation Models

Cupertino, California

•

Today

Join the team shaping the data foundation and intelligence for Apple's frontier foundation models. We believe that breakthrough AI capabilities are driven not only by model architecture and scale, but by the quality, diversity, and intelligence of the data used to train them. As part of the Apple Foundation Model team, you will help define how next-generation foundation models learn, reason, plan, and interact with the world, powering intelligent experiences used by billions of people. This is a

Full-time

Evaluation & Insights Machine Learning Engineer

Cupertino, California

•

Today

Imagine what you could do here. At Apple, great new ideas have a way of becoming extraordinary products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish! Are you passionate about music, movies, and the world of Artificial Intelligence and Machine Learning? So are we! Join our Human-Centered AI team for Apple Products. In this role, you'll represent the user perspective on new features, review and analyze

Full-time

On-device ML Performance Engineer, Graphics, Games and Machine Learning

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 On-device ML Performance team has the responsibility to analyze latency, memory, p

Full-time

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

Jax jobs in San Jose, CA, USA