JAX Jobs in San Jose, CA

Refine Results
1 - 20 of 44 Jobs

Senior Software Engineer, JAX

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

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. Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX. To achieve this and provide an excellent user experience, you have strong programming skills and excellent
Posted 1 day ago | Updated 10 hours ago

Senior Performance Engineer - Deep Learning

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both inside and outside of NVIDIA across fields, as well as collaborate with the open-source community to
Posted 1 day ago | Updated 10 hours ago

Senior Performance Engineer - Deep Learning

NVIDIA

•

San Jose, California, USA

Full-time

NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both inside and outside of NVIDIA across fields, as well as collaborate with the open-source community to
Posted 41 days ago | Updated 7 hours ago

Engineering Manager, Generative AI

NVIDIA Corporation

•

Santa Clara, California, 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.
Posted 8 days ago | Updated 10 hours ago

Senior Research Engineer for Reinforcement Learning

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is searching for a senior or principal engineer who specializes in large-scale reinforcement learning and policy learning in the Generalist Embodied Agent Research (GEAR) group. Our team is leading Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-scale robot learning, embodi
Posted 32 days ago | Updated 10 hours ago

Software Engineer - Robot Learning Platform

NVIDIA Corporation

•

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
Posted 60+ days ago | Updated 10 hours ago

Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is searching for a senior or principal engineer who specializes in building cutting-edge infrastructure for large-scale foundation model training in the Generalist Embodied Agent Research (GEAR) group. Our team is leading Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-sca
Posted 32 days ago | Updated 10 hours ago

Senior Performance Engineer - Deep Learning

NVIDIA

•

Fremont, California, USA

Full-time

NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both inside and outside of NVIDIA across fields, as well as collaborate with the open-source community to
Posted 41 days ago | Updated 7 hours ago

Senior Performance Engineer - Deep Learning

NVIDIA

•

Palo Alto, California, USA

Full-time

NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both inside and outside of NVIDIA across fields, as well as collaborate with the open-source community to
Posted 41 days ago | Updated 15 hours ago

AI Research Engineer, Data Quality, Tesla Bot

Tesla Motors

•

Palo Alto, California, USA

Full-time

Tesla is on a path to build humanoid robots at scale to automate repetitive and boring tasks. The goal of our Machine Learning team is to build and demonstrate a general robot learning system that can leverage AI to perform complex physical tasks, ranging from full body locomotion, and precise manipulation This specific role will do hands-on data analysis for Tesla Bot's Machine Learning team, as well as build data pipelines, tools and applications to automate those analysis. This is a unique op
Posted 23 days ago | Updated 10 hours ago

Distinguished Engineer, AI Resiliency Lead

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are seeking a Distinguished Engineer to lead AI Resiliency at NVIDIA! Join NVIDIA and help push the boundaries of AI. In this role, you will architect, design, and develop world-class software resiliency features for training ground breaking AI models on the largest AI superclusters in the world. Leading a team of cross-functional experts, you will drive and shape our end-to-end AI software stack, ensuring seamless training of frontier models on industry-leading frameworks like PyTorch and J
Posted 60+ days ago | Updated 10 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 60+ days ago | Updated 10 hours ago

AI Research Engineer, Data Scaling, Self-Driving

Tesla Motors

•

Palo Alto, California, USA

Full-time

At Tesla, you will have access to unparalleled resources that set us apart from other companies in the AI industry. With the world's largest self-driving dataset, you will have a unique opportunity to develop and optimize a large-scale data engine that powers our autonomous driving systems. Tesla's extensive fleet of vehicles generates massive amounts of real-world data daily, giving you unprecedented access to diverse driving scenarios and edge cases. Additionally, Tesla offers one of the highe
Posted 58 days ago | Updated 10 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 60+ days ago | Updated 10 hours ago

Staff Software Engineer, Generative AI Platforms

Tesla Motors

•

Palo Alto, California, USA

Full-time

This Software Engineer will be at the forefront of developing & delivering cutting-edge applications on Tesla's advanced generative AI platform. Your primary focus will be on building & enhancing core components of the generative AI system, ensuring they are both efficient and scalable. You will collaborate closely with team members to translate complex research ideas into robust, production-ready services. This role requires a strong foundation in modern machine learning frameworks and a proven
Posted 60+ days ago | Updated 10 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 49 days ago | Updated 10 hours ago

Technical Marketing Engineer - AI Platform Software

NVIDIA

•

Santa Clara, California, USA

Full-time

NVIDIA software powers today's breakthroughs in AI! To enable researchers and developers to keep pace with this dynamic field, we seek a technical marketing expert who understands the AI platform software stack and the ecosystem! This role will craft the fundamental technical content educating developers how to write software's using NVIDIA's AI platforms through technical blog posts, user guides, walk-throughs, benchmark and more. This material is essential in mentoring developers about the lat
Posted 60+ days ago | Updated moments ago

Senior Site Reliability Engineer, AI Infrastructure

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you! NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for over 30 years. It's a unique legacy of innovation that's fueled by phenomenal technology and outstanding people. Today, we're tapping into the unlimited potential of
Posted 21 hours ago | Updated 10 hours ago

Sr. Machine Learning Engineer, AI Infrastructure

Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Machine Learning Engineer on the Bottle Rocket team, you will play a critical role in developing innovative, data-driven solutions on Tesla's generative AI platform. You will focus on leveraging machine learning models and deep learning techniques to solve complex problems and deliver impactful insights. Collaborating with cross-functional teams, you'll translate research concepts into scalable data products and drive data-centric strategies for optimizing AI applications. Your work will en
Posted 31 days ago | Updated 10 hours ago

Software Engineering Manager - 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 math libraries in the AI, HPC, and Quantum Computing (QC) domains. Around the world, leading commercial and academic organizations are revolutionizing AI, scientific and engineering simulations, and data analytics, using data centers powered by GPUs. Applications of these technologies include LLMs, Computer Vision, Encryption, autonomous vehicles and countless others. Did you know our team develops the GPU acce
Posted 1 day ago | Updated 10 hours ago