Design and evaluate MLOps engineering tasks for frontier AI systems using JAX, PyTorch, and custom GPU kernels with Pallas or Triton while improving ML infrastructure, distributed training, and AI model evaluation.
Remote job listings
Find your next remote opportunity from thousands of listings across the globe.
Work on advanced JAX engineering tasks involving compiler internals, custom transformations, distributed sharding, XLA/MLIR integration, and high-performance machine learning systems for frontier AI research.
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code.
Conduct cutting-edge LLM research spanning foundation model pre-training, large-scale data curation, post-training, alignment, and evaluation for leading frontier AI labs.
JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast. NVIDIA supercomputers are becoming more heterogeneous combining GPUs, CPUs, and LPUs, and JAX needs to evolve to perform well across all of them.
NVIDIA's invention of the GPU 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing.
Join NVIDIA to help university researchers advance the next generation of foundation models, multimodal AI, reasoning systems, and AI agents! At NVIDIA, we build accelerated computing platforms for frontier AI research.
Join a leading AI lab's cutting-edge Generative AI team and help build foundational AI models from the ground up. This opportunity is seeking experienced MLOps Engineers with deep expertise in JAX and kernel-level programming using Pallas or Triton. The role focuses on AI…
NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services.
DGX Station (Galaxy) is NVIDIA’s workstation-class AI computer—built on GB300 Blackwell GPUs with NVLink interconnect, delivering data-center-grade AI compute in a deskside form factor.
NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services.
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.
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.
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…
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Are you passionate about blazing the unbounded field of deep learning and building the best software tools for developers ?
We're looking for outstanding AI systems engineers to develop groundbreaking technologies in the inference systems software stack! We build innovative AI systems software to accelerate for AI inference.
We're looking for outstanding AI systems engineers to develop groundbreaking technologies in the inference systems software stack! We build innovative AI systems software to accelerate for AI inference.
Join a leading AI lab’s GenAI team and contribute to the development of foundational large language models and AI infrastructure systems. This opportunity is seeking experienced MLOps Engineers with strong expertise in: Machine learning infrastructure Distributed training…
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
As a Member of Technical Staff on Machine Learning, you will: Contribute to the entire development cycle of our cutting-edge large deep learning models. Prepare datasets, design architectures, implement solutions, train and evaluate models to improve our products.
Tips for finding remote jobs
- Set up job alerts on multiple platforms to never miss an opportunity.
- Highlight your remote work experience and self-management skills.
- Prepare for video interviews and remote work assessments.
- Customize your resume and cover letter for each remote position.
- Build a strong online presence on LinkedIn and professional networks.