JAX Jobs in Virginia

Refine Results
1 - 20 of 30 Jobs

Senior AI Engineer [ No Corp-to-Corp submittals ]

PMC

•

Remote

Contract

This is not a C2C position. 6 Months with Extensions Must Haves: Strong Python GitHub Unix EXCELLENT understanding of Networking Understands Containers and Orchestration laC and Cloud Deployment Strong Sql Solid Engineering background in the workplace. AI/MLOps Engineer Must answer questions during phone screen. The AI/MLOps Engineer (AMOE) is an applied solutions expert who combines working knowledge of software engineering, machine learning, and DevOps best practices. As part of a small tea
Easy Apply
Posted 10 days ago

Spring Boot Developer

Delmock Technologies, Inc

•

Remote or Lanham, Maryland, USA

Full-time

About Our Company: Delmock Technologies, Inc. (DTI), is a leading HUBZone business in Baltimore, known for delivering sophisticated IT (Information Technology) and Health solutions with a commitment to ethics, expertise, and superior service. Actively engaged in the local community, DTI creates opportunities for talented residents while maintaining a stellar reputation as an award-winning contractor, earning accolades like the Government Choice Award for IRS (Internal Revenue Service) Systems Mo
Posted 13 days ago | Updated 11 hours ago

Machine Learning Engineer (Life Sciences, Bio Medical, Bio Informatics, BioTech)

Projas Technologies, LLC

•

Remote

Contract

Machine Learning EngineerAbout the Role: We are seeking a highly motivated and skilled Machine Learning Engineer to join our growing team.As a Machine Learning Engineer, you will play a crucial role in developing and implementing cutting-edge machine learning algorithms and systems to revolutionize the drug discovery process.This role offers a unique opportunity to work on challenging problems with the potential to significantly impact patient lives.Prior experience in Research and Development r
Easy Apply
Posted 57 days ago | Updated 22 days ago

Senior Software Engineer, 3D Computer Vision

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is a world-leader in high speed computer vision, artificial intelligence, and deep learning. Our team builds the accelerated software ecosystem that enables visual AI developers to innovate swiftly and efficiently at scale. We are now looking for an outstanding CUDA developer to build developer-facing libraries and services that will accelerate the development of autonomous vehicles, humanoid robots, XR, and manufacturing. The right candidate will combine deep experience with the 3D appl
Posted 2 days ago | Updated 2 hours ago

Deep Learning Engineer - Distributed Task-Based Backends

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are looking for Senior to Principal level experienced software professionals to help build the next generation of distributed backends for premier Deep Learning frameworks like PyTorch, JAX and TensorFlow. You will build on top of validated task-based runtime systems like Legate, Legion & Realm to develop a platform that can scale a wide range of model architectures to thousands of GPUs! What You Will Be Doing: Develop extensions to popular Deep Learning frameworks, that enable easy experime
Posted 45 days ago | Updated 2 hours ago

Senior Software Engineer, Computer Vision Microservices

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is a world-leader in high speed computer vision, artificial intelligence, and deep learning. Our team builds the accelerated software ecosystem that enables visual AI developers to innovate swiftly and efficiently at scale. Are you looking for a technical leadership role building high-performance distributed AI? We are seeking an outstanding individual to help us build highly optimized microservice products and NVIDIA NIMs that bring visual AI to large scale data. The right candidate will
Posted 14 days ago | Updated 2 hours ago

Software Engineer, Deep Learning - Autonomous Vehicles

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are looking for outstanding Deep Learning Software Engineers to develop and productize NVIDIA's deep learning solutions in autonomous driving vehicles. We work on innovative vision and language models to ensure they are accelerated by NVIDIA SW. In addition, we contribute to our open-source library and have opportunities to present our work at conferences. As a member of our Solution Engineering-Automotive Deep Learning team, you will apply ground breaking NVIDIA deep learning model training/
Posted 45 days ago | Updated 2 hours ago

Senior Software Engineer - Automated Parallel Programming

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

The PyTorch Team @ NVIDIA is hiring passionate parallel programmers. Join us to design and build the tools used by millions of AI practitioners deploying AI applications scalable to thousands of GPUs. Our team is responsible for the continual delivery of best in class experience on NVIDIA's hardware with PyTorch. Join our team and collaborate with many multi-disciplinary engineering teams within NVIDIA and internationally in the PyTorch open source community to deliver our customers the best of
Posted 45 days ago | Updated 2 hours ago

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
Posted 45 days ago | Updated 2 hours ago

Senior Deep Learning Software Engineer - Developer Tools

NVIDIA Corporation

•

Remote or Westford, Massachusetts, USA

Full-time

Are you passionate about blazing the unbounded field of deep learning and building the best software tools for developers? We are looking for an outstanding software engineer to join the Developer Tools organization at NVIDIA that seeks to deliver the best-in-class tool to enable streamlined design and optimization of high-performance deep neural network models in a unified environment. Come be part of our team! What You Will Be Doing: Innovate on tools support for design and performance optimi
Posted 45 days ago | Updated 2 hours ago

Senior Artificial Intelligence Algorithms Engineer

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Artificial Intelligence 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 AI /Deep-Learning(DL) and a deep algorithmic background, with exposure to computer architecture and performance, 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 an
Posted 45 days ago | Updated 2 hours ago

Senior Deep Learning Software Engineer, Algorithmic Model Optimization

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Deep Learning Software Engineer, for Algorithmic Model Optimization! Join our team of algorithmic model optimization experts and take part in unlocking the biggest potential for AI with generative models such as large language models (LLM) and diffusion models. As a Senior Deep Learning Software Engineer, you will be at the forefront of pushing the boundaries of these models and enabling their deployment at a larger scale with unmatched efficiency. We are develop
Posted 43 days ago | Updated 2 hours ago

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
Posted 45 days ago | Updated 2 hours ago

Senior AI Software Engineer, GenAI Framework

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for AI Software Engineers for NeMo! NVIDIA NeMo is an open-source, scalable and cloud-native framework built for researchers and developers working on Large Language Models (LLM), Multimodal (MM), and Speech AI. NeMo provides end-to-end model training, including data curation, alignment, customization, evaluation, deployment and tooling to optimize performance and user experience. In this role you will craft and implement new model development features, optimizations, define A
Posted 45 days ago | Updated 2 hours ago

Software Engineering Manager, Linear Algebra Libraries

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are looking for a software engineering manager to lead our development efforts for high-performance linear algebra libraries in the AI and scientific computing domains. Around the world, leading commercial and academic organizations are revolutionizing AI, data analytics, and scientific and engineering simulations, using data centers powered by GPUs. Applications of these technologies include LLMs, computer vision, encryption, computer aided engineering (CAE), electronic design automation (ED
Posted 45 days ago | Updated 2 hours ago

Senior HPC Performance Engineer - AI for Science at Scale

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

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
Posted 45 days ago | Updated 2 hours ago

Senior Scientific Machine Learning Software Engineer - Physics

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA's deep learning and HPC platforms have made a huge impact in various fields and are broadly used across leading academic institutions, start-ups, and industry, including the world's largest Internet companies. We need passionate and creative people to help us on building a AI framework that will solve the toughest and most relevant problems of humanity and problems that are at the groundbreaking of science & engineering: weather/climate challenges, product design, digital twins, molecular
Posted 25 days ago | Updated 2 hours ago

Senior Software Engineer - Python Numerical Computing Libraries

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are looking for an experienced software professional to contribute to design and development of accelerated and distributed implementations of Python APIs for numerical computing. In the last decade, Python has become the de-facto programming language for practitioners in AI, data science and HPC, through popular frameworks such as NumPy, SciPy, TensorFlow and PyTorch. These frameworks provide an efficient high-level programming interface, allowing their users to focus on their application wh
Posted 26 days ago | Updated 2 hours ago

Machine Learning Engineering Graduate Internship - Summer 2025

CVS Health

•

Remote or New York, New York, USA

Full-time

You've invested a lot of time and energy in your education. Now you want the chance to make your mark. We offer challenging opportunities for you to learn and grow professionally. In our programs, you'll be immersed in a culture of continuous improvement, with the goal of changing health care for the better. Position Summary This program is a 11-week full-time opportunity that provides real work experience with a team in your business area. You will be aligned to projects to lead or contribute
Posted 18 days ago | Updated 3 hours ago

Lead Data Scientist, NLP & Strategic Member Insights

CVS Health

•

Connecticut, USA

Full-time

Bring your heart to CVS Health. Every one of us at CVS Health shares a single, clear purpose: Bringing our heart to every moment of your health. This purpose guides our commitment to deliver enhanced human-centric health care for a rapidly changing world. Anchored in our brand - with heart at its center - our purpose sends a personal message that how we deliver our services is just as important as what we deliver. Our Heart At Work Behaviors support this purpose. We want everyone who works at C
Posted 45 days ago | Updated 3 hours ago