JAX jobs in Arizona

Research Fellow - Data Science and Artificial Intelligence

Scottsdale, Arizona

•

Today

Job Description A postdoctoral research fellow position is available at the Cancer Data Science Lab (PI: Dr. Irbaz Riaz - Dr. Irbaz B. Riaz, Bio) at Mayo Clinic in Arizona. The research fellow will work in a multi-disciplinary team of clinicians, data scientists and informaticians. Dr. Riaz is an Assistant professor in medicine and biomedical informatics. His lab is focused on developing artificial intelligence (AI) methods for electronic medical records data. The current projects heavily emph

Full-time

Applied ML Scientist

Remote

•

17d ago

A globally leading technology company is looking for an Applied ML Scientist to design and evaluate next-generation machine learning and GenAI systems that power real-world applications. In this role, you ll develop advanced LLM-based evaluation frameworks, prototype AI-driven solutions, and collaborate closely with cross-functional teams to bring cutting-edge research into production. If you re passionate about building impactful ML systems and driving innovation at scale, we invite you to appl

Easy Apply

Contract

$100 - $120

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

Solutions Architect, AI Models

Remote or Santa Clara, California

•

Today

Do you want to be part of the team that brings innovative Artificial Intelligence (AI) from research to reality? We are looking for a Solution Architect to join the NVIDIA AI Enterprise (NVAIE) SA Segment team. We specialize in the newest technology and advances in deep learning, Generative AI, and Cloud. The vision of the NVAIE Segment team is to use our deep expertise, at the intersection of research and engineering, to guide and enable the successful adoption of NVIDIA AI software in the ente

Full-time

USD 148,000.00 per year

Senior Solutions Architect, Generative AI

Remote or Santa Clara, California

•

Today

NVIDIA is looking for an AI Solutions Architect with hands-on experience in efficient AI model training and/or deployment for a customer facing role. Primary responsibilities will be to help accelerate customer workloads and lead customer technical engagements around NVIDIA software and technologies. We work with the world's most successful technology companies, and this role uniquely positions you to observe emerging industry trends at the cutting edge. Join us in this exciting endeavor! What

Full-time

USD 184,000.00 per year

DGX Cloud Performance Engineer

Remote or Santa Clara, California

•

Today

NVIDIA DGX Cloud is an end-to-end, scalable AI platform for developers, offering scalable capacity built on the latest NVIDIA architecture and co-engineered with the world's leading cloud service providers (CSPs). We are seeking highly skilled Parallel and Distributed Systems engineers to drive the performance analysis, optimization, and modeling to define the architecture and design of Nvidia's DGX Cloud clusters. The ideal candidate will have a deep understanding of the methodology to conduc

Full-time

USD 184,000.00 per year

Research Scientist/Engineer L5/L6, AI for Member Systems

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. At Netflix, our mission is to entertain the world. That means connecting billions of people with movies, TV shows, and games they'll love. To do this, we invest in deep

Full-time

USD 230,000.00 per year

Senior Math Libraries Engineer, CPU and GPU Optimization

Remote or Santa Clara, California

•

Today

NVIDIA is looking for an expert software engineer to help us deliver CUDA-X libraries across the NVIDIA CPU and GPU ecosystem. For over a decade, NVIDIA's accelerated computing platform has revolutionized HPC and AI with applications ranging from COVID-19 research to autonomous machines. Did you know that our team develops the GPU/CPU-accelerated mathematical libraries that make all of this possible? The hardware and software accelerated computing ecosystem is constantly evolving, including shi

Full-time

USD 224,000.00 per year

Senior Software Engineer, Compute Infrastructure for Robotics Research

Remote or Santa Clara, California

•

Today

We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This new team will work very closely with NVIDIA's Generalist Embodied Agent Research (GEAR) group. The near term focus is Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. This particular position will focus on compute infrastructure. You will work with an amazing and collaborative research team that

Full-time

USD 148,000.00 per year

Lead Software Engineer - AI

Remote or Frisco, Texas

•

Today

Lead Software Engineer, AI Are you passionate about building AI-driven software that transforms how professionals work? Join a dynamic and highly skilled team at Thomson Reuters, where we invest deeply in AI technologies and explore emerging fields with the backing of a global leader. As a Staff Software Engineer - AI, you will play a lead role in developing intelligent systems that power the next generation of expert solutions across legal, tax, risk, and compliance domains. About The Role A

Full-time

USD 147,000.00 - 273,000.00 per year

Senior Deep Learning Software Engineer - Autonomous Vehicles

Remote or Santa Clara, California

•

Today

We are looking for outstanding Deep Learning Software Engineers to develop and productize NVIDIA's deep learning solutions in autonomous driving vehicles. As a member of our Solution Engineering-Automotive Machine Learning team, you will apply ground breaking NVIDIA deep learning model training/inference software libraries for deployment on NVIDIA's hardware architecture. You will develop new deep learning architectures, train deep learning models, and compile and optimize DNN graphs. As a part

Full-time

USD 148,000.00 per year

Senior Software Engineer, MathDx - cuBLASDx and cuSolverDx

Remote or Santa Clara, California

•

Today

NVIDIA is looking for an expert software engineer to help us expand our catalog of Device eXtension (Dx) APIs for our math libraries. For over a decade, NVIDIA's accelerated computing platform has revolutionized HPC and AI with applications ranging from COVID-19 research to autonomous machines. Did you know that our team develops the GPU-accelerated mathematical libraries that make all of this possible? The hardware and software accelerated computing ecosystem is constantly evolving, including

Full-time

USD 148,000.00 per year

Senior Deep Learning Algorithm 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 184,000.00 per year

Senior Research Engineer

Remote or Santa Clara, California

•

Today

Join NVIDIA and help build the software that will define the future of generative AI. We are looking for a research engineer who is passionate about open-source and excited to create our next-generation post-training software stack. You will work at the intersection of research and engineering, collaborating with the Post-Training and Frameworks teams to invent, implement, and scale the core technologies behind our Nemotron models. What you'll be doing: Work with applied researchers to design,

Full-time

USD 160,000.00 per year

Senior Generative AI Research Engineer

Remote or Santa Clara, 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 and post-train foundation models (LLMs, VLMs,

Full-time

USD 224,000.00 per year

Senior Deep Learning Software Engineer, Inference and Model Optimization

Remote or Santa Clara, California

•

Today

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

Full-time

USD 148,000.00 per year

Machine Learning Engineer 5 - Content & Studio

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. At Netflix, we seek to entertain the world. We launch thousands of new TV shows and movies every year for our members across the globe. To do so, we need to understand

Full-time

USD 150,000.00 per year

Senior Software Engineer, Data Infrastructure for Robotics Research

Remote or Santa Clara, California

•

Today

We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This new team will work very closely with NVIDIA's Generalist Embodied Agent Research (GEAR) group. The near term focus is Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. This particular position will focus on data infrastructure. You will work with an amazing and collaborative research team that con

Full-time

USD 224,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

Senior MLOps Engineer

Remote or Santa Clara, California

•

Today

NVIDIA is seeking a Senior MLOps Engineer to help design and scale the infrastructure that powers our AI research and product development. In this role, you will partner closely with research scientists and product teams to accelerate their success on the latest GPU and accelerator platforms. By building robust ML pipelines and scalable systems, you will ensure that cutting-edge hardware innovations translate directly into faster experiments, more efficient training, and reproducible deployments

Full-time

USD 184,000.00 per year

Filter Results

Job post features

Posted date

Work settings

Employment type

Employer type

Work authorization

JAX jobs in Arizona, USA