JAX Jobs in Seattle, WA

Refine Results
1 - 20 of 21 Jobs

Machine Learning AI Engineer 2

Apex Systems

Redmond, Washington, USA

Full-time

Job#: 2080130 Job Description: Location Requirements: Hybrid in Redmond, WA - 3 days a week onsite. Flexible on days. Our team seeks a Machine Learning Engineer to advance the development and improvement of our software foundation and tools vital for training state-of-the-art AI models. Your role will be centered on creating strong, scalable, and efficient training infrastructures and frameworks facilitating the full spectrum of the machine learning process, from handling data to deploying mode

Senior Machine Learning Engineer

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 Deep Learning Compiler Engineer - CUDA

NVIDIA Corporation

Seattle, 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.

Software Engineer - Model Training Infrastructure - USDS

TikTok

Seattle, Washington, USA

Full-time

Location : Seattle Employment Type : Regular Job Code : A35774 Apply to this job Share this listing: Responsibilities About the team The mission of our AML team is to push the next-generation AI infrastructure and recommendation platform for the ads ranking, search ranking, live & ecom ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Machine Learning Engineer - Model Training Infrastructure to join our team to suppor

Software Engineer - Model Serving Infrastructure - USDS

TikTok

Seattle, Washington, USA

Full-time

Location : Seattle Employment Type : Regular Job Code : A32426A Apply to this job Share this listing: Responsibilities Team Intro The mission of our AML team is to push the next-generation AI infrastructure and recommendation platform for the ads ranking, search ranking, live & ecom ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Machine Learning Engineer - Model Serving Infrastructure to join our team to support an

Machine Learning Scientist

RWE Renewables

Bellevue, Washington, USA

Full-time

RWE Supply & Trading Americas, LLC To start as soon as possible, fulltime, permanent Your future plans We are building our core AI Weather Lab team to explore creative applications of AI methods to challenges in the renewable energy industry. To help us innovate and build industry-leading AI models, we are seeking an experienced Machine Learning Scientist with a strong mathematical and research background in diffusion models, transformerbased models, or related areas. As a key member of our r

Director Software Development - AI Inference

AMD (Advanced Micro Devices)

Bellevue, Washington, USA

Full-time

WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellen

Principal Software Engineer, TensorRT-LLM

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

Staff Software Engineer, Machine Learning

Discord Inc.

Remote or San Francisco, California, USA

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

Developer Technologies Engineer, Robotics Reinforcement Learning

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 Engineer

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

Principal Deep Learning Software Engineer, LLM Performance

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

Senior Deep Learning Software Engineer, Inference and Model Optimization

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

Senior Deep Learning Software Engineer, LLM Performance

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior 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 in

(USA) Senior, Data Scientist

Walmart Inc.

Remote or New York, New York, USA

Full-time

Position Summary Join Walmart|VIZIO and take your career to the next level! We are looking for a highly talented and motivated Data Scientist to help deliver strategic initiatives across the organization. We look for people who can effectively partner with Product Management, Engineering, and business leaders to deliver easy, effective, and insightful analytics. We are a growing team, focusing on challenging problems that make a difference to our business. We concentrate on high-impact, high-va

Senior DGX Cloud AI Infrastructure Software Engineer

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

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

Machine Learning Software Engineer L4/L5

Netflix, Inc.

Remote

Full-time

Netflix is one of the world's leading entertainment services, with 283 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. As Netflix continues to grow, so do the opportunities to enhance our personalization systems and algorithms. We're looking for a passionate and talented Software Engineer fo

Senior Math Libraries Engineers - Python APIs

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

NVIDIA is looking for a self-motivated and specialist software engineer for the design and development of high-performant Python APIs for our math libraries. In the last decade, Python has become the de-facto leading programming language for engineers in AI and data science, and more recently in HPC and scientific computing. NVIDIA has been at the forefront of providing GPU accelerated Deep Learning frameworks. These frameworks provide an efficient high-level programming interface allowing their

Software Engineer Staff

Lockheed Martin Corporation

Remote or Colorado Springs, Colorado, USA

Full-time

Job Description Space is a critical domain, connecting our technologies, our security and our humanity. While others view space as a destination, we see it as a realm of possibilities, where we can do more - we can innovate, invest, inspire and integrate our capabilities to transform the future. At Lockheed Martin Space, we aim to harness the full potential of space to cultivate innovation, reduce costs, and push the boundaries of what technology can achieve. We're creating future-ready solutio

AI Data Engineer

Randstad Digital

Redmond, Washington, USA

Contract

job summary: The primary consideration for this position is a preference for candidates with experience in robotics or reinforcement learning, rather than those specializing in natural language models or computer vision location: Redmond, Washington job type: Contract salary: $67 - 77 per hour work hours: 8am to 5pm education: Bachelors responsibilities: - Create and uphold efficient, scalable, and distributed training systems-including data preprocessing, training orchestration, and mode