Explore AI roles · Updated daily

Search curated engineering roles across AI infrastructure, agents, machine learning, and product teams.

Keyboard friendly Save and share roles

We are now looking for a Senior Deep Learning Software Engineer, PyTorch-TensorRT Performance! NVIDIA is seeking an e...

Fulltime Deep Learning Pytorch Generative Ai

$148k* · Senior Fully remote

View details

AI Software Engineer (Generative AI) Azumo , a leading AI Development company , is seeking a highly motivated AI Soft...

Fulltime Remote Ai Data Science

Fully remote

View details

We are now looking for a Senior Deep Learning Software Engineer, PyTorch. NVIDIA is hiring software engineers to desi...

Fulltime Ai Ai Engineer Deep Learning

$100k* · Senior Fully remote

View details

About Gridware Gridware is a San Francisco-based technology company dedicated to protecting and enhancing the electri...

Fulltime Machine Learning Ai

Fully remote

View details

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its s...

Fulltime Remote Machine Learning Robotics

$204k - $259k · Senior Fully remote

View details

Who is Flock? Flock Safety is the leading safety technology platform, helping communities thrive by taking a proactiv...

Fulltime Remote Machine Learning Mlops

Fully remote

View details

Title: Applied AI Robotics Engineer Location: Remote Duration: 12+ months Required Qualifications PhD in a relevant S...

Contract Remote Ai Machine Learning

Fully remote

View details

We are seeking passionate Machine Learning Engineers to join our Inference team, focusing on the creative application...

Fulltime Ai Machine Learning Generative Ai

Fully remote

View details

Vectara provides a scalable platform to deploy your Enterprise AI Agents and AI Assistants with Accuracy, Security, a...

Fulltime Remote Ai Machine Learning

Fully remote

View details

Vectara provides a scalable platform to deploy your Enterprise AI Agents and AI Assistants with Accuracy, Security, a...

Fulltime Remote Ai Machine Learning

Fully remote

View details

The base salary for this position is dependent upon experience and location. Locations: Orange, CT or Boston, MA - (I...

Ai Machine Learning Data Science Nlp

$125k - $164k · Senior Fully remote

View details

A company that is building the infrastructure that powers next-generation multi-modal AI research is looking for a nu...

Fulltime Ai Machine Learning Pytorch

Fully remote

View details

Secure Global Money Transfers with Cutting-Edge Technology. Join our mission to protect cross-border transactions, he...

Remote Ai Machine Learning Data Science

Fully remote

View details

• Develop AI solutions and intelligent agents tailored to automotive use cases. • Automated generation of new test ca...

Contract Ai Machine Learning Data Science

$14k* · Senior Fully remote

View details

Architect scalable, high-performance AI/ML solutions for predictive maintenance, and real-time analytics. Develop AI ...

Fulltime Remote Ai Machine Learning

Fully remote

View details

PENN Entertainment, Inc. is North America’s leading provider of integrated entertainment, sports content, and casino ...

Fulltime Machine Learning Data Science Mlops

$103k - $161k · Senior Fully remote

View details

We are seeking a "GenAI and Data Architect" with experience in Snowflake Cortex Platform. This is a Contract and Remo...

Contract Remote Ai Machine Learning

$135k - $156k · Senior Fully remote

View details

The AI Frontiers lab at Microsoft Research is charted with ambitious research goals for advancing Artificial Intellig...

Fulltime Ai Generative Ai Pytorch

$100k - $199k · Senior Fully remote

View details

Join the Siri Planner team as an experienced Machine Learning Research Engineer and play a pivotal role in shaping th...

Ai Machine Learning Nlp Generative Ai

Fully remote

View details

Are you passionate about Generative AI? Are you interested in working on groundbreaking generative modeling technolog...

Ai Machine Learning Computer Vision Generative Ai

$13k* · Senior Fully remote

View details

Selected role

Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance

NVIDIA Santa Clara, CA

$148k Senior Full-time

We are now looking for a Senior Deep Learning Software Engineer, PyTorch-TensorRT Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate about analyzing and improving the performance of Torch inference with TensorRT! 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 areas like Generative AI, Recommenders and Vision that have put DL into every software solution. Join the team that builds the software to enable the performance optimization, deployment and serving of these DL solutions. We specialize in developing GPU-accelerated Deep learning software like TensorRT, DL benchmarking software and performant solutions to deploy and serve these models.

Collaborate with the deep learning community to integrate TensorRT to PyTorch. Identify performance opportunities and optimize SoTA models across the spectrum of NVIDIA accelerators, from datacenter GPUs to edge SoCs. Implement graph compiler algorithms, frontend operators and code generators across the PyTorch, Torch-TensorRT, TensorRT software stack. Work and collaborate with a diverse set of teams involving workflow improvements, performance modeling, performance analysis, kernel development and inference software development.

What You'll Be Doing

Analyze performance issues and identify performance optimization opportunities inside Torch-TensorRT/TensorRT.
Contribute features and code to NVIDIA/OSS inference frameworks including but not limited to Torch-TensorRT/TensorRT/PyTorch.
Work with cross-collaborative teams inside and outside of NVIDIA across generative AI, automotive, robotics, image understanding, and speech understanding to develop innovative inference solutions.
Scale performance of deep learning models across different architectures and types of NVIDIA accelerators.

What We Need To See

Bachelors, Masters, PhD, or equivalent experience in relevant fields (Computer Science, Computer Engineering, EECS, AI).
At least 4 years of relevant software development experience.
Excellent Python/C++ programming, software design and software engineering skills
Experience with a DL framework like PyTorch, JAX, TensorFlow.
Experience with performance analysis and performance optimization

Ways To Stand Out From The Crowd

Architectural knowledge of GPU.
Prior experience with a AoT or JiT compiler in deep learning inference, e.g. TorchDynamo/TorchInductor.
Prior experience with performance modeling, profiling, debug, and code optimization of a DL/HPC/high-performance application.
GPU programming experience and proficiency in one of the GPU programming domain specific languages, e.g. CUDA/TileIR/CuTeDSL/cutlass/Triton.

GPU deep learning has provided the foundation for machines to learn, perceive, reason and solve problems posed using human language. The GPU started out as the engine for simulating human imagination, conjuring up the amazing virtual worlds of video games and Hollywood films. Now, NVIDIA's GPU runs deep learning algorithms, simulating human intelligence, and acts as the brain of computers, robots and self-driving cars that can perceive and understand the world. Just as human imagination and intelligence are linked, computer graphics and artificial intelligence come together in our architecture. Two modes of the human brain, two modes of the GPU. This may explain why NVIDIA GPUs are used broadly for deep learning, and NVIDIA is increasingly known as “the AI computing company.” Come, join our DL Architecture team, where you can help build the real-time, cost-effective computing platform driving our success in this exciting and quickly growing field.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 148,000 USD - 235,750 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.

You will also be eligible for equity and benefits .

Applications for this job will be accepted at least until December 15, 2025.NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

JR2009866

Showing 11401–11420 of 14,332 AI roles

1 … 569 570 572 573 717

Find your next AI role

Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance

AI Software Engineer (Generative AI) - Remote

Senior Software Engineer, PyTorch - Deep Learning

Senior ML Infrastructure Engineer

Senior Software Engineer, Simulation ML Infrastructure

Senior Software Engineer, ML Platform

Applied AI Robotics Engineer

Generative AI Inference Engineer

Machine Learning (ML) Engineer

Machine Learning (ML) Engineer

Lead Architect - Generative AI

Machine Learning Research Engineer

Senior DevOps / ML Infrastructure Engineer - AI Lab

Generative AI /Machine Learning Test Automation Engineer

AI/ ML Architect with IoT Experience - Remote

Software Engineer, ML Platform

Generative AI Architect

Research Engineer - Generative AI - AI Frontiers

Sr. Machine Learning Research Engineer - Large Language Models (LLMs), Siri Core Modeling

Generative AI Research Engineer, Multimodal, Agent Modeling - SIML