ZipRecruiter

Log In

1

Pytorch Developer Jobs in California (NOW HIRING)

CYNET SYSTEMS

Java AI Developer

Palo Alto, CA · On-site

$53 - $57/hr

Seeking a Java AI Developer with strong programming skills and experience in cloud services to ... Familiarity with AI/ML frameworks (TensorFlow, PyTorch) and experience with generative AI tools (e ...

Qualcomm

Sr. AI Engineer

Santa Clara, CA · On-site

$122K - $168K/yr

Qualcomm Technologies, Inc. is seeking a skilled and motivated AI Model Training Engineer to join ... PyTorch, onnxruntime or Hugging Face Transformers. • Proficient in Python and familiar with ML ...

Toyon

Deep Learning Algorithm Developer

Goleta, CA

$120K - $200K/yr

Toyon has openings for researchers and developers to solve challenging real-world problems using ... Experience with PyTorch, TensorFlow, or other deep learning frameworks is required. An advanced ...

previous pagenext page

Showing results 21-40

All JobsPytorch Developer JobsPytorch Developer Jobs in California

Pytorch Developer information

What is a PyTorch developer?

A PyTorch Developer is a software engineer or data scientist who specializes in using PyTorch, an open-source machine learning library, to build and deploy deep learning models. Their responsibilities typically include designing neural network architectures, training and evaluating models, and optimizing code for performance. PyTorch Developers work in fields such as artificial intelligence, computer vision, and natural language processing, collaborating with teams to solve complex problems using machine learning. They are proficient in Python and have a strong understanding of deep learning concepts. Additionally, they often contribute to research, development, and the deployment of AI solutions in production environments.

What are some common challenges PyTorch developers face when deploying machine learning models to production environments?

Pytorch Developers often encounter challenges when transitioning models from research to production, such as optimizing model performance for inference speed and memory usage, ensuring compatibility with deployment frameworks like TorchScript or ONNX, and managing dependencies across different systems. Additionally, integrating PyTorch models into existing software stacks and maintaining reproducibility can be complex. Collaborating closely with DevOps and data engineering teams is crucial to address these issues and ensure smooth deployment.

What are the key skills and qualifications needed to thrive as a PyTorch developer, and why are they important?

To thrive as a Pytorch Developer, you need strong programming skills in Python, a solid grasp of machine learning concepts, and experience with deep learning frameworks—especially PyTorch itself. Familiarity with tools like CUDA, Jupyter Notebooks, and version control systems (e.g., Git) is typically expected, along with knowledge of cloud platforms or relevant certifications. Problem-solving ability, effective collaboration, and clear communication are crucial soft skills for success in this role. These skills and qualities are vital for efficiently building, optimizing, and deploying machine learning models in real-world applications.

What is the difference between Pytorch Developer vs Machine Learning Engineer?

AspectPytorch DeveloperMachine Learning Engineer
Required CredentialsBachelor's or higher in CS, experience with PyTorchBachelor's or higher in CS, data science, or related field, with ML experience
Work EnvironmentResearch labs, AI startups, tech companies focusing on deep learningTech companies, finance, healthcare, often involving deployment and scaling ML models
Industry UsagePrimarily in AI research and development teamsAcross industries implementing ML solutions in production

While both roles require knowledge of machine learning and experience with PyTorch, a Pytorch Developer mainly focuses on developing and optimizing deep learning models using PyTorch. A Machine Learning Engineer often has a broader scope, including deploying, maintaining, and scaling ML models across various platforms and industries.

What are popular job titles related to Pytorch Developer jobs in California?

For Pytorch Developer jobs in California, the most frequently searched job titles are:

What job categories do people searching Pytorch Developer jobs in California look for?

The top searched job categories for Pytorch Developer jobs in California are:

What cities in California are hiring for Pytorch Developer jobs?

Cities in California with the most Pytorch Developer job openings:

Pytorch Developer jobs near you
Infographic showing various Pytorch Developer job openings in California as of August 2026, with employment types broken down into 79% Full Time, 5% Part Time, 2% Temporary, and 14% Contract. Highlights an 80% Physical, 6% Hybrid, and 14% Remote job distribution.

Principal Developer, AI Networking

Nvidia

Santa Clara, CA • On-site

Full-time

Re-posted 5 days ago

Nvidia rating

9.6

Company rating: 9.6 out of 10

Based on 17 frontline employees who took The Breakroom Quiz

7th of 244 rated software companies

Job description

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. As an NVIDIAN, you'll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

The AI Networking Codesign and Benchmarking R&D group requires a senior software engineer. In this exciting role, you will profile, analyze, and optimize AI workloads on large-scale GPU and CPU clusters used for distributed Deep Learning LLM training and inference. Your primary focus will be collectives communication and networking. You will work across hardware components such as HCAs, Switches, CPUs, GPUs, and Systems. You will also engage with software layers including LLM applications, machine learning frameworks, communication, and computing libraries. Moreover, you will build performance analysis tools and strategies to investigate details and clarify performance expectations, limitations, and bottlenecks. This is your chance to contribute to AI innovation!

What you'll be doing:

  • Characterizing AI workloads and deep learning models aimed at large-scale LLM training and inference on NVIDIA supercomputers. The role centers on distributed systems with a focus on high-performance networking and NVIDIA communication libraries.

  • Benchmarking, profiling, and analyzing the performance to find bottlenecks and identify areas for improvement and optimizations, with a strong emphasis on networking aspects.

  • Developing PyTorch trace-based profiling, analysis, and replaying toolset to aid in benchmarking, debugging, and co-designing network systems for LLM workloads.

  • Collaborating with multiple teams from hardware to software to provide performance analysis insights.

  • Defining performance test plans, setting performance expectations for new technologies and solutions, and working to achieve performance targets.

What we need to see:

  • B.Sc in Computer Science or Software Engineering or equivalent experience.

  • 15+ years of experience with high-performance networking (RDMA, MPI, NCCL, SHARP).

  • Demonstrated ability in performance evaluation techniques and approaches.

  • Experience with NVIDIA GPUs and the CUDA library. Knowledge of deep learning frameworks like TensorFlow or PyTorch. Expertise in networking collective communication libraries such as NCCL and protocols like RoCE and RDMA.

  • Fast and self-learning capabilities with strong analytical and problem-solving skills.

  • Proficiency in programming languages: Python, Bash, and C++.

  • Experience with a container-based development environment.

  • Great teammate who communicates clearly and works well with others.

Ways to stand out from the crowd:

  • Extensive understanding and hands-on experience with AI workloads and benchmarking for distributed LLM training.

  • Knowledge in PyTorch, CUDA, and NCCL libraries.

  • Comprehensive system knowledge and understanding (Intel / AMD / ARM CPUs, NVIDIA GPUs, HCA, Memory, PCI).

  • Strong capabilities in performance evaluation and methods using contemporary tools.

With competitive salaries and a generous benefits package, we are 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 and, due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 272,000 USD - 431,250 USD for Level 6, and 320,000 USD - 488,750 USD for Level 7.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until June 16, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive 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.

What Nvidia employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom

Nvidia logo

About Nvidia

Sourced by ZipRecruiter

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.

Industry

Computer and electronic product manufacturing

Company size

10,000+ Employees

Headquarters location

Santa Clara, CA, US

Year founded

1993

View All Nvidia Jobs

Apply