JAX jobs in Boston, MA

Senior Speech Machine Learning Engineer, Siri

Cambridge, Massachusetts

•

Today

We are a group of engineers/researchers responsible for advancing Siri Conversational AI at Apple. Our mission is to build cutting-edge infrastructure, datasets, and models that empower Siri with powerful general capabilities across natural language understanding, dialog generation, speech recognition, and multi-modal interaction. We apply these technologies to create engaging, intelligent, and personalized conversational experiences for millions of Apple users! We believe that the most impactfu

Full-time

USD 162,500.00 - 286,400.00 per year

Senior Speech Machine Learning Engineer

Cambridge, Massachusetts

•

Today

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

Full-time

USD 162,500.00 - 286,400.00 per year

Lead Research Engineer

Boston, Massachusetts

•

Today

Anduril Industries is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st century's most innovative companies to the defense industry, Anduril is changing how military systems are designed, built and sold. Anduril's family of systems is powered by Lattice OS, an AI-powered operating system that turns thousands of data streams into a realtime, 3D command and c

Full-time

USD 198,000.00 - 342,000.00 per year

Ray Inference Engineer

Remote

•

Today

Position Summary: Designing, implementing, and maintaining distributed systems to build world-class ML platforms/products at scaleExperiment with, deploy, and manage LLMs in a production contextBenchmark and optimize inference deployments for different workloads, e.g. online vs. batch vs. streaming workloadsDiagnose, fix, improve, and automate complex issues across the entire stack to ensure maximum uptime and performanceDesign and extend services to improve functionality and reliability of the

Easy Apply

Contract

$80 - $90

Staff Software Engineer, Machine Learning

Remote or San Francisco, California

•

Today

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

Full-time

USD 272,000.00 - 306,000.00 per year

GenAI Architect/ Director

Remote

•

Today

Position : GenAI Architect/ Director (AWS AI services) Location: San Diego, CA / Remote Job Type: Contract Required Skills : AWS AI services (Bedrock, SageMaker, etc.) Modern AI frameworks (PyTorch, TensorFlow, JAX) LLM(GPT, Claude, Llama) Retrieval-Augmented Generation (RAG) MLOps practices Multimodal AI systems (text, image, audio) Job Description 10+ years of experience in data science, machine learning, or AI fields 5+ years in leadership roles related to AI governance, strategy, or innova

Easy Apply

Full-time, Part-time, Contract, Third Party

Principal Software Engineer, TensorRT-LLM

Remote or Santa Clara, California

•

Today

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

Full-time

USD 272,000.00 per year

Senior Cloud Services Software Engineer

Remote or Santa Clara, California

•

Today

Joining NVIDIA's DGX Cloud Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of ML workloads, as well as developing scalable AI infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the vital resources and scale to champion innovation. We are seeking a distributed software engineer to join our team! As a Senior engineer, yo

Full-time

USD 184,000.00 per year

Machine Learning Engineer (L5) - Content & Media ML Foundations

Remote

•

Today

Netflix is one of the world's leading entertainment services, with over 300 million paid memberships in over 190 countries enjoying TV series, films and games across a wide variety of genres and languages. Members can play, pause and resume watching as much as they want, anytime, anywhere, and can change their plans at any time. The Content & Media ML Foundations team within Data Science and Engineering, builds and delivers high-leverage foundational ML solutions using Netflix's unique media da

Full-time

USD 170,000.00 per year

Senior Generative AI Research Engineer

Remote or California

•

Today

At NVIDIA, we're not just building the future, we're generating it. Our Cosmos generative AI engineering team is pushing the boundaries of what's possible across multimodal learning, video generation, synthetic data, intelligent simulation, and agentic systems. We are looking for exceptionally driven engineers and applied scientists with deep experience in generative modeling to help define the next era of AI computing. What you'll be doing: Design, post-train, and optimize foundation models (e

Full-time

USD 224,000.00 per year

Machine Learning Data Engineer

Remote or Hybrid in Atlanta, Georgia

•

5d ago

We are seeking a highly skilled LLM Engineer with expertise in Google Cloud Platform (Google Cloud Platform), Vertex AI, and algorithm development to design, implement, and optimize cutting-edge AI solutions. You will work on building, fine-tuning, and deploying Large Language Models (LLMs) for real-world business applications. Key Responsibilities Design, develop, and optimize algorithms for LLM training, inference, and deployment on Google Cloud Platform. Implement Vertex AI pipelines for data

Easy Apply

Third Party, Contract

$80 - $100

Senior Software Engineer, AI Storage Infrastructure

Remote or Santa Clara, California

•

Today

NVIDIA's invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern deep learning - the next era of computing - with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as "the AI computing company." We're looking to grow our company and establish teams with the mos

Full-time

USD 168,000.00 per year

Principal Artificial Intelligence Algorithms Engineer

Remote or Santa Clara, California

•

Today

NVIDIA is looking for engineers for our core AI Frameworks (Megatron Core and NeMo Framework) team to design, develop and optimize diverse real world workloads. Megatron Core and NeMo Framework are open-source, scalable and cloud-native frameworks built for researchers and developers working on Large Language Models (LLM) and Multimodal (MM) foundation model pretraining and post-training. Our GenAI Frameworks provide end-to-end model training, including pretraining, reasoning, alignment, customi

Full-time

USD 272,000.00 per year

Senior HPC Performance Engineer - AI for Science at Scale

Remote or Santa Clara, California

•

Today

NVIDIA has become the platform upon which every new AI-powered application is built. We are seeking a Sr. HPC Performance engineer to join our team of scientists and engineers passionate about building the next generation of scientific machine learning (ML) frameworks. Starting with digital biology, through high performance computing (HPC) and powerful ML methods, together, we will advance NVIDIA's capacity to accelerate AI for Science and industries that depend on it. What you'll be doing: De

Full-time

USD 184,000.00 per year

Lead Data Scientist, NLP & Strategic Member Insights

Remote or Connecticut

•

Today

At CVS Health, we're building a world of health around every consumer and surrounding ourselves with dedicated colleagues who are passionate about transforming health care. As the nation's leading health solutions company, we reach millions of Americans through our local presence, digital channels and more than 300,000 purpose-driven colleagues - caring for people where, when and how they choose in a way that is uniquely more connected, more convenient and more compassionate. And we do it all w

Full-time

USD 130,295.00 - 260,590.00 per year

Lead Data Scientist - Autonomous Goal Management

Remote

•

Today

Become a part of our caring community and help us put health first The Enterprise AI organization at Humana is a pioneering force, driving AI innovation across our Insurance and CenterWell business segments. By collaborating with world-leading experts, we are at the forefront of delivering cutting-edge AI technologies for improving care quality and experience of millions of consumers. We are actively seeking top talent to develop robust and reusable AI modules and pipelines, ensuring adherence t

Full-time

USD 142,300.00 - 195,700.00 per year

Lead Data Scientist - Safety Alignment

Remote

•

Today

Full-time

USD 142,300.00 - 195,700.00 per year

Senior Applied Machine Learning Scientist - Recommendations

Remote or Lake Forest, Illinois

•

Today

Work Location Type: Hybrid About Grainger: W.W. Grainger, Inc., is a leading broad line distributor with operations primarily in North America, Japan and the United Kingdom. At Grainger, We Keep the World Working by serving more than 4.5 million customers worldwide with products and solutions delivered through innovative technology and deep customer relationships. Known for its commitment to service and award-winning culture, the Company had 2024 revenue of $17.2 billion across its two busines

Full-time

USD 121,500.00 - 202,500.00 per year

GPU SME- 10+ yrs- Remote (US Only)

Remote

•

15d ago

Job description: GPU / TPU Experience Familiarity with hands on IaC is a must Terraform GKE Networking Storage Python Library familiarity with: numpy/pandas/Pytorch/JAX, including optimization Experience with Nvidia and/or Google TPU hardware in GCE and GKE ML specific Google Cloud Platform products: Parallel store, Hyperdisk ML, TCP Direct Troubleshooting and optimization of large (1000s of nodes) GKE clusters

Easy Apply

Contract, Third Party

Up to $65

Lead Data Scientist - AI-Human Interaction

Remote

•

Today

Full-time

USD 142,300.00 - 195,700.00 per year

Filter Results

Job post features

Posted date

Work settings

Employment type

Distance

Employer type

Work authorization

JAX jobs in Boston, MA, USA