1

Geometric Deep Learning Jobs in California (NOW HIRING)

Background in differentiable optimization and geometric deep learning approaches. Experience optimizing ML models for mobile deployment and resource-constrained environments. Knowledge of SLAM ...

Data Scientist

Torrance, CA · On-site

$170K - $300K/yr

Geometric deep learning: mesh / point-cloud networks, GNNs, PyTorch Geometric * CAD / B-rep, feature recognition, and turning part geometry into ML features * Retrieval and ANN at scale; embedding ...

Data Scientist

Los Angeles, CA · On-site

$170 - $300/hr

Geometric deep learning: mesh / point‑cloud networks, GNNs, PyTorch Geometric * CAD / B‑rep, feature recognition, and turning part geometry into ML features * Retrieval and ANN at scale ...

Data Scientist

Torrance, CA · On-site

$170 - $300/hr

Geometric deep learning: mesh / point-cloud networks, GNNs, PyTorch Geometric * CAD / B-rep, feature recognition, and turning part geometry into ML features * Retrieval and ANN at scale; embedding ...

Data Scientist

Santa Monica, CA · On-site

$170 - $300/hr

Geometric deep learning: mesh / point-cloud networks, GNNs, PyTorch Geometric * CAD / B-rep, feature recognition, and turning part geometry into ML features * Retrieval and ANN at scale; embedding ...

Data Scientist

Los Angeles, CA · On-site

$170K - $300K/yr

Geometric deep learning: mesh / point-cloud networks, GNNs, PyTorch Geometric * CAD / B-rep, feature recognition, and turning part geometry into ML features * Retrieval and ANN at scale; embedding ...

Head of ML

San Francisco, CA · On-site

$250K - $450K/yr

Background in machine learning for 3D perception - point cloud understanding, 3D detection/segmentation, geometric deep learning, or related areas. * Experience with CAD AI, design automation, or ...

ML on raw experimental data rather than processed structures * 3D vision and geometric deep learning backgrounds especially welcome Dataset generation and open release * Designing and running ...

Showing results 21-40

Geometric Deep Learning information

What is the difference between Geometric Deep Learning vs Data Scientist?

AspectGeometric Deep LearningData Scientist
Required CredentialsAdvanced degrees in computer science, machine learning, or related fieldsBachelor's or master's in data science, statistics, or related fields
Work EnvironmentResearch labs, AI development teams, academiaBusiness analytics, product teams, consulting firms
Industry UsageAI, robotics, computer vision, graph analysisBusiness intelligence, marketing, finance, healthcare

Geometric Deep Learning focuses on applying deep learning techniques to non-Euclidean data like graphs and manifolds, often requiring advanced technical skills. Data Scientists analyze and interpret data to inform business decisions, typically working with structured data and statistical tools. While both roles involve data analysis, Geometric Deep Learning is more research-oriented and specialized in AI development, whereas Data Scientists focus on practical data insights across industries.

What are some common challenges faced when working on geometric deep learning projects, and how can they be addressed?

One common challenge in Geometric Deep Learning is dealing with the complexity and diversity of data structures, such as graphs, point clouds, or manifolds. These data types often require specialized neural network architectures and custom preprocessing steps, which can be more complex than traditional deep learning tasks. Collaboration with domain experts and staying updated with the latest research are crucial for overcoming these obstacles. Additionally, debugging and visualizing the learning process can be more challenging, so employing robust evaluation metrics and visualization tools is highly recommended.

What are the key skills and qualifications needed to thrive as a geometric deep learning engineer?

To excel as a Geometric Deep Learning Engineer, you need a strong background in mathematics, machine learning, and computer science, typically supported by an advanced degree in a related field. Proficiency with deep learning frameworks like PyTorch or TensorFlow, as well as experience with graph neural networks (GNNs) and geometric data structures, is essential. Strong analytical thinking, problem-solving abilities, and collaborative communication are key soft skills for innovating and working with interdisciplinary teams. These skills are crucial for developing cutting-edge models that leverage geometric data, enabling impactful solutions across domains such as computer vision, biology, and social network analysis.

What are popular job titles related to Geometric Deep Learning jobs in California?

For Geometric Deep Learning jobs in California, the most frequently searched job titles are:

What job categories do people searching Geometric Deep Learning jobs in California look for?

The top searched job categories for Geometric Deep Learning jobs in California are:

What cities in California are hiring for Geometric Deep Learning jobs?

Cities in California with the most Geometric Deep Learning job openings:

Infographic showing various Geometric Deep Learning job openings in California as of August 2026, with employment types broken down into 100% Full Time. Highlights an 87% In-person, and 13% Remote job distribution.

Senior Software Engineer -- cuEquivariance

NVIDIA

Santa Clara, CA • On-site

$143K - $189K/yr

Full-time

Re-posted 25 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

Job Summary:
NVIDIA has been a leader in computer graphics and accelerated computing for over 25 years and is now leveraging AI for the next computing era. They are seeking a Senior Software Engineer to join the cuEquivariance team, responsible for building and optimizing GPU-accelerated geometric ML primitives and collaborating with research teams to deliver production-quality software for scientific applications.
Responsibilities:
• Build, implement, and optimize CUDA kernels for equivariant neural network primitives — tensor products, segmented polynomials, and triangle-based operations — targeting peak performance across NVIDIA GPU generations.
• Be responsible for the end-to-end delivery of GPU-accelerated geometric ML primitives: from implementation to validated, production-quality software that external frameworks depend on.
• Build and maintain the interfaces for PyTorch and JAX that expose cuEquivariance primitives to application developers and researchers.
• Drive CI/CD infrastructure for multi-GPU kernel builds, automated correctness testing, and performance regression tracking.
• Collaborate with Applied Science and research teams to evaluate new equivariant architectures and translate prototypes into production kernels.
• Engage directly with third-party framework developers and partners to align on interfaces and ensure delivered software integrates cleanly into production pipelines.
Qualifications:
Required:
• 6+ years of software engineering experience with a strong background in CUDA and GPU programming.
• Deep proficiency in C++ and Python; experience building and shipping production libraries used by external developers.
• Good foundation in GPU computing: memory hierarchy, warp-level execution, occupancy, and performance profiling methodology.
• Experience building or chipping in to production scientific software libraries, ML frameworks, or developer-facing GPU APIs.
• Familiarity with concepts in geometric machine learning — equivariance, group representations, irreducible representations, or tensor products — sufficient to work efficiently in the domain.
• BS/MS in Computer Science, Physics, Applied Mathematics, or a related field, or equivalent experience.
Preferred:
• You have chipped in to or deeply used a major neural network framework that respects equivariance: e3nn, MACE, NequIP, SE(3)-Transformers, or similar.
• Hands-on experience with Triton kernel development or other GPU kernel authoring tools alongside CUDA.
• Experience with mixed-precision or tensor-core-aware algorithm design for scientific or ML workloads.
• PhD or equivalent experience in computational chemistry, biophysics, physics, or computer science with a focus on geometric deep learning or HPC.
• Contributions to open-source geometric ML or GPU computing projects.
Company:
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. Founded in 1993, the company is headquartered in Santa Clara, USA, with a team of 10001+ employees. The company is currently Late Stage.

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