Find jobs in AI/ML, Data Science and Big Data

7 results for Torch compile (Skill/Tech stack)

Machine Learning Engineer II - Autonomous Driving & Inference Runtime USD 180K-210K

Batching | C# | C++ | CUDA | FP16

Dental insurance | Disability insurance | Flexible spending account | Flexible vacation | Health insurance

Mid-level Full Time

Anywhere, USA R

1d ago
AI Inference Performance Engineer - New College Grad 2026 USD 124K-241K

Attention | Batching | C++ | CUDA | CUDA kernels

Benefits | Equity

Senior-level Full Time

US, CA, Santa Clara, United States

7d ago
Staff Compiler Engineer - PyTorch + Kernel DSLPLATE USD 163K-253K

Autotuning | Collective Primitives | Cost Based Compilation | Custom ISA | Cutlass

401k | Adoption support stipend | Charitable giving match | Fertility care stipend | Flexible work environment

Senior-level Full Time

San Jose, California, United States

7d ago
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356K

C++ | CUDA | CUDA kernel | CUDA kernel optimization | Computer Architecture

Equity | Health benefits | Paid time off

Senior-level Full Time

US, CA, Santa Clara, United States

27d ago
Member of Technical Staff, Performance Optimization USD 175K-220K

CUDA | CUPTI | Distributed Systems | GPU Profiling | Infiniband

Senior-level Full Time

San Mateo, CA

28d ago
Senior ML Inference Engineer - Platform USD 128K-261K

Abstractions | Airflow | CUDA | Flyte | Inference Integration

Employee assistance program | Health insurance benefits | Life insurance | Medical/Dental/Vision | Paid vacation

Senior-level Full Time

GM Automation - Sunnyvale - GM …

1mo ago
AI Engineer - Model Performance USD 165K-250K

Attention Backend | Audio Processing | Batching | CUDA | CUDA graph

Async communication | Innovation-focused culture | Remote work | Startup environment | Supportive team

Mid-level Full Time

SF Hybrid R

1mo ago