JAX Jobs

21 - 40 of 176 Jobs

Senior Machine Learning EngineerJob Title - Senior Machine Learning Engineer

Randstad - TorcCompany Name - Randstad - Torc

•

Remote

Contract

We are looking for a Senior Machine Learning Engineer to join our team and drive the development of advanced AI-driven solutions. You will be responsible for designing, deploying, and optimizing machine learning models that power real-world applications. This role requires deep expertise in model architecture, data engineering, MLOps, and production-scale AI systems. What you'll do:Design, develop, and deploy scalable, production-ready machine learning models for real-world applications.Architec

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

Machine Learning Engineer - NYC- Leading Global Prop Trading FirmJob Title - Machine Learning Engineer - NYC- Leading Global Prop Trading Firm

Oxford KnightCompany Name - Oxford Knight

•

New York, New York, USA

Full-time

Salary: up to $300,000 USD base + annual bonus Summary Exciting opportunity to work at one of the world's leading propriety trading firms with offices across the globe. You will be working with an elite global team of top minds in machine learning and engineering to help drive the direction of their ML platform. Seeking accomplished engineers with expertise in machine learning and robust mathematical foundations to work on building the firm-wired machine learning infrastructure and/or working wi

Developer Technologies Engineer, Robotics Reinforcement LearningJob Title - Developer Technologies Engineer, Robotics Reinforcement Learning

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

Today, NVIDIA is 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 never been done before takes vision, innovation, and the world's best talent. As an NVIDIAN, you'll be immersed in a diverse, encouraging environment where everyone is inspired to do their best work. Come join the team and see how we can make a lasting impact on the world

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

AIML - Machine Learning Researcher, Foundation ModelsJob Title - AIML - Machine Learning Researcher, Foundation Models

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

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

On-device ML Infrastructure Engineer (ML Compiler Frontend)Job Title - On-device ML Infrastructure Engineer (ML Compiler Frontend)

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 Speech Machine Learning EngineerJob Title - Senior Speech Machine Learning Engineer

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Want to join the team pushing the boundaries of AI and building an intelligent assistant that helps millions of people get things done? Join the Siri team at Apple. To build the best speech recognition models, we need to use the latest technology in distributed training and the best available data. We combine those needs into one team and are focused on blurring the lines between traditional "data processing" and "model training". Efficiently training on petabytes of audio data pushes us to cons

On Device Machine LearningJob Title - On Device Machine Learning

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The On-Device Machine Learning team at Apple enables 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 field, collaborating with research, SW engineering, HW engineering, and products. The team builds critical infrastructure that begins with onboarding the latest machine learning architectures to

Senior Compiler Verification Engineer - Deep LearningJob Title - Senior Compiler Verification Engineer - Deep Learning

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are building the next generation of compiler technologies to accelerate deep learning workloads. We are looking for an engineer to work on compiler verification 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 developers to verify new and state of the a

Principal Deep Learning Software Engineer, LLM PerformanceJob Title - Principal Deep Learning Software Engineer, LLM Performance

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Deep Learning Software Engineer, LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate about analyzing and improving the performance of LLM inference! NVIDIA is rapidly growing our research and development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join our team. Companies around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs

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

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

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. This amazing team is responsible for enabling the Research to Production lifecycle of cutting edge ML models that power magical user experiences on Apple's hardware and software platforms. We build critical infrastructure that begins with onboarding the latest machine learning architectures to embedded devices, opti

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

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

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

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Senior Deep Learning Software Engineer, Inference and Model OptimizationJob Title - Senior Deep Learning Software Engineer, Inference and Model Optimization

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is at the forefront of the generative AI revolution! The Algorithmic Model Optimization Team specifically focuses on optimizing generative AI models such as large language models (LLM) and diffusion models for maximal inference efficiency using techniques ranging from neural architecture search and pruning to sparsity, quantization, and automated deployment strategies. Our work includes conducting applied research to improve model efficiency as well as developing an innovative software pl

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

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

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 - 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 EngineerJob Title - AIML - Machine Learning Engineer

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time