jax Jobs

21 - 40 of 168 Jobs

Staff Engineer, Machine Learning - Online MappingJob Title - Staff Engineer, Machine Learning - Online Mapping

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 capit

AIML - Sr. Machine Learning Engineer, Search Quality - Answers, Knowledge & Intelligence (AKI)Job Title - AIML - Sr. Machine Learning Engineer, Search Quality - Answers, Knowledge & Intelligence (AKI)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

We are looking for a hardworking and experienced Machine Learning Engineer to build intelligent search experiences. In this role, you will build intelligent search systems that deeply understand user intent and context to return highly relevant, accurate, and personalized responses. You will develop and deploy machine learning models to shape how users find relevant content or answers, driving better experiences and higher satisfaction! Description As a member of our fast-paced group, you'll hav

On-device ML Infrastructure Engineer (ML Insights and Forecasting)Job Title - On-device ML Infrastructure Engineer (ML Insights and Forecasting)

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 innovative 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 area, working with research, SW engineering, HW engineering, and products. The team builds critical infrastructure that begins with onboarding the latest machine learning arc

AIML - Machine Learning EngineerJob Title - AIML - Machine Learning Engineer

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

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

On-device ML Performance Infrastructure EngineerJob Title - On-device ML Performance Infrastructure Engineer

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 for analyzing latency, memory and numerical cor

Model Optimization Engineer (PyTorch Infrastructure Development)Job Title - Model Optimization Engineer (PyTorch Infrastructure Development)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Are you excited about the impact that optimizing deep learning models can have on enabling transformative user experiences? The field of ML compression research continues to grow rapidly and new techniques to perform quantization, pruning etc are increasingly available to be ported and adopted by the ML developer community, that is looking to ship more models in a constrained memory budget and make them run faster. We are passionate about productizing and pushing the envelope of the state of the

Senior Machine Learning Engineer - Large Language Models (LLMs), Siri Planner TeamJob Title - Senior Machine Learning Engineer - Large Language Models (LLMs), Siri Planner Team

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Join the Siri Planner team as an experienced Machine Learning Engineer and play a pivotal role in shaping the future of virtual assistants. You'll contribute to revolutionizing how millions of Siri users worldwide interact with their Apple devices by building ground breaking AI solutions. Description As a Senior Machine Learning Engineer, you will lead initiatives to significantly advance Siri's natural language understanding and planning capabilities using innovative LLM technologies. YOUR CORE

AIML - Staff ML System Engineer, ML Platform Technologies (MLPT)Job Title - AIML - Staff ML System Engineer, ML Platform Technologies (MLPT)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Want to build the machine learning platform that engineers rely on to develop next-generation Apple Intelligence products and services? As a machine learning engineer on our team, you will create platforms and tools to enable performant, scalable ML workload for Apple's AI-driven experiences. Join our team of highly skilled, impact-focused engineers! This role also includes opportunities to open source your work and publish at top ML conferences. Description We're searching for strong machine le

AIML - Machine Learning Engineer, Siri and Information IntelligenceJob Title - AIML - Machine Learning Engineer, Siri and Information Intelligence

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Principal Software Engineer, TensorRT-LLMJob Title - Principal Software Engineer, TensorRT-LLM

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Software Engineer, TensorRT-LLM ! NVIDIA is hiring experienced principal software engineer for its TensorRT-LLM team. Academic and commercial groups around the world are using GPUs to power a revolution in AI, enabling breakthroughs in areas like content creation, code generation and information synthesis, that have put AI on the precipice of an "iPhone moment". Join the team building the AI serving software which is foundational to product lines within NVIDIA

Software Engineer, Machine Learning, Sensor SimulationJob Title - Software Engineer, Machine Learning, Sensor Simulation

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Software Engineer, Machine Learning, 3D ReconstructionJob Title - Software Engineer, Machine Learning, 3D Reconstruction

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior Software Engineer, Machine Learning, 3D ReconstructionJob Title - Senior Software Engineer, Machine Learning, 3D Reconstruction

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior Deep Learning Software EngineerJob Title - Senior Deep Learning Software Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are looking for a Senior Deep Learning Software Engineer to design and build our automated inference and deployment solution. As part of the team, you will be instrumental in defining a scalable architecture for DL inference with emphasis on ease-of-use and compute efficiency. Your work will span multiple layers of the DL deployment stack, encompassing developing features in high-level frameworks like PyTorch and JAX, designing and implementing a high-performance execution environment, low-le

Senior GenAI Algorithms EngineerJob Title - Senior GenAI Algorithms Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Gen AI Algorithms Engineer! NVIDIA is seeking engineers to design, develop and optimize Artificial Intelligence solutions to diverse real-world problems. If you have a strong understanding of deep learning and in particular large language models and their multimodal variants, then this role may be a great fit for you! Collaborate and interact with internal partners, users, and members of the open source community to analyze, define and implement highly optimized A

Staff Software Engineer, Machine LearningJob Title - Staff Software Engineer, Machine Learning

Discord Inc.Company Name - Discord Inc.

•

Remote

Full-time

Discord is used by over 200 million people every month for many different reasons, but there's one thing that nearly everyone does on our platform: play video games. Over 90% of our users play games, spending a combined 1.5 billion hours playing thousands of unique titles on Discord each month. Discord plays a uniquely important role in the future of gaming. We are focused on making it easier and more fun for people to talk and hang out before, during, and after playing games. We're currently l

AIML - Staff Machine Learning Engineer, Siri and Information IntelligenceJob Title - AIML - Staff Machine Learning Engineer, Siri and Information Intelligence

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

AIML - ML EngineerJob Title - AIML - ML Engineer

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

As part of Apple's Machine Learning Research organization, we do world-class scientific research and build the technologies that will power future products at Apple. The techniques and tools we create will impact ML solutions across Apple, which in turn power most of the features we deliver to billions of consumers worldwide. We are looking for highly motivated, result-oriented engineers with a strong background in Computer Science, Machine Learning engineering and software development to join o

Software Engineer, Machine Learning, CalibrationJob Title - Software Engineer, Machine Learning, Calibration

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior Software Development Engineer, AI/ML, AWS Neuron, Model InferenceJob Title - Senior Software Development Engineer, AI/ML, AWS Neuron, Model Inference

Annapurna Labs (U.S.) Inc.Company Name - Annapurna Labs (U.S.) Inc.

•

Cupertino, California, USA

Full-time

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machinelearning accelerators and servers that use them. This role is for a software engineer in the Machine Learning Inference Model Enablement team for AWS Neuron at Annapurna Labs.This role is responsible for development, enablement and performance tuning of a wide variety of LLM model families, including massive scale large language models like the Llama family, DeepSeek and beyond, as well as stable dif