JAX Jobs

21 - 40 of 129 Jobs

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

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

Computer Vision Software EngineerJob Title - Computer Vision Software Engineer

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The Video Computer Vision organization is working on exciting technologies for future Apple products. Our team delivers computer vision and machine learning algorithms that power many Apple technologies like human understanding and human intelligence algorithms with applications for digital humans, health and AI. In this role, you will work closely with our team of experts in computer graphics, computer vision and machine learning to design and build solutions to previously unsolved challenges f

Software AI EngineerJob Title - Software AI Engineer

Selby JenningsCompany Name - Selby Jennings

•

Boston, Massachusetts, USA

Full-time

About the Firm We are a world-class investment management firm headquartered in Boston, known for applying advanced technology and data-driven approaches to generate superior returns across global markets. Our culture emphasizes intellectual rigor, innovation, and collaboration. We are building next-generation systems that leverage artificial intelligence to inform investment decisions, streamline operations, and unlock new research capabilities. The Opportunity We're seeking a highly skilled S

Senior Software Engineer, AWS Neuron InferenceJob Title - Senior Software Engineer, AWS Neuron Inference

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

•

Seattle, Washington, USA

Full-time

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machinelearning accelerators. This role is for a senior software engineer in the Machine Learning Inference Applications team. This role is responsible for development and performance optimization of core building blocks of LLM Inference - Attention, MLP, Quantization, Speculative Decoding, Mixture of Experts, etc.The team works side by side with chip architects, compiler engineers and runtime engineers to

Senior Software Engineer, AWS Neuron InferenceJob Title - Senior Software Engineer, AWS Neuron Inference

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

•

Seattle, Washington, USA

Full-time

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators. This role is for a senior software engineer in the Machine Learning Inference Applications team. This role is responsible for development and performance optimization of core building blocks of LLM Inference - Attention, MLP, Quantization, Speculative Decoding, Mixture of Experts, etc. The team works side by side with chip architects, compiler engineers and runtime engineers

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

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

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

Uber CorporateCompany Name - Uber Corporate

•

Sunnyvale, California, USA

Full-time

About the Role We are looking for an experienced Senior Machine Learnign Engineer to join the Trusted Identity engineering team at Uber. The Trusted Identity org plays a crucial role in our mission to thoroughly verify the identity of our customers (riders, eaters) and partners (drivers, restaurants) while ensuring powerful, intuitive and streamlined user experiences. Our continued obsession for innovation in the space is essential to ensure safety as people interact on Uber's platform. The Ac

Software Engineer- AI/ML, AWS NeuronJob Title - Software Engineer- AI/ML, AWS Neuron

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

•

Seattle, Washington, USA

Full-time

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machinelearning accelerators and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive scale large language models like LLama3, Mixtral, DBRX and beyond, as well as stable diffusion, V

Software Engineer- AI/ML, AWS NeuronJob Title - Software Engineer- AI/ML, AWS Neuron

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

•

Seattle, Washington, USA

Full-time

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive scale large language models like LLama3, Mixtral, DBRX and beyond, as well as stable diffusion,

On-device ML Infrastructure Engineer (Fine-tuning and Customization)Job Title - On-device ML Infrastructure Engineer (Fine-tuning and Customization)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

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

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Are you excited about Generative AI and Large Language Models? Are you interested in working on cutting edge generative modeling technologies and its application to virtual assistants to enrich billions of people!? Join the AI/ML Response team at Apple. Our team is responsible for developing cutting-edge machine learning technology for natural language and speech generation, including projects like Apple Intelligence and Siri. We are looking for exceptional senior applied research engineer exper

AI/ML EngineerJob Title - AI/ML Engineer

JobotCompany Name - Jobot

•

Chicago, Illinois, USA

Full-time

AI/ML Engineer need for well-established SaaS company This Jobot Job is hosted by: Joseph Calabrese Are you a fit? Easy Apply now by clicking the "Apply Now" button and sending us your resume. Salary: $130,000 - $160,000 per year A bit about us: We are transforming the financial sponsor universe through cutting-edge data science and generative AI solutions. Creating an ecosystem of connected data and actionable intelligence that solves strategic business challenges with purposeful innovation

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

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

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

Computer Vision & Machine Learning EngineerJob Title - Computer Vision & Machine Learning Engineer

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 a 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 Appl

AIML - Senior ML Engineer, Generative AI - Siri Comprehension & PlanningJob Title - AIML - Senior ML Engineer, Generative AI - Siri Comprehension & Planning

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 innovative 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, opening boundaries with the latest generative AI research developments, and ultimately delivering product critical models that power Siri and Apple Inte

Machine Learning Engineer, Apple Services EngineeringJob Title - Machine Learning Engineer, Apple Services Engineering

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Are you passionate about building extraordinary products and embracing state of art techniques? It's central to all that we do! The goal of the Apple Services GenAI & ML Frameworks team is to take personalization, search, and more to the next level of intelligence and accuracy using advanced ML and statistical techniques. Description We work on a wide spectrum of approaches, including foundation modeling, deep learning, model compression, transfer learning, federated learning and more. We work o

Full Stack .NET DeveloperJob Title - Full Stack .NET Developer

JobotCompany Name - Jobot

•

Tulsa, Oklahoma, USA

Contract

Hybrid Contract Dev Opportunity This Jobot Consulting Job is hosted by: Kyle Miller Are you a fit? Easy Apply now by clicking the "Apply Now" button and sending us your resume. Salary: $110,000 - $140,000 per year A bit about us: At Jobot, we are disrupting the recruiting and staffing space by: Using the latest AI technology to match jobs to job seekers. Having a culture built on kindness and respect. Hiring experienced recruiters who believe in providing the best possible service to their

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

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Redmond, Washington, USA

Full-time

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 never been done before takes vision, innovation, and the world's best talent.