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

8 results for Torch compile (Skill/Tech stack)

Generative AI - ML System Engineering USD 175K-300K

BF16 | C++ | CUDA | Data loading | Data-parallel

Comprehensive benefits package | Growth opportunities | In-office work | Silicon Valley work environment

Senior-level Full Time

Bay Area Office

5d ago
LLM Engineer, Agentic Researcher Platform USD 152K-287K

ASHA | Automl | Bayesian optimization | Benchmarking | CUDA

Mid-level Full Time

Vietnam, Ho Chi Minh City

7d ago
Staff MLOps Engineer USD 190K-272K

AWS | Azure | BentoML | C++ | CI/CD

Senior-level Full Time

Onsite - Austin, TX

21d ago
Senior Deep Learning Framework Communications Engineer USD 152K-287K

C++ | CUDA | CUDA kernels | CuTe | Distributed Training

Benefits | Equity

Senior-level Full Time

US, CA, Santa Clara R

1mo ago
Senior Machine Learning Engineer - Research Optimisation AUD 121K-161K

A/B | A/B Testing | AWS | B testing | Batching

Flexible leave options | Hybrid remote work | Inclusive parental leave | Wellbeing allowance

Senior-level Full Time

Sydney, Australia R

1mo ago
LLM Inference Frameworks and Optimization Engineer USD 160K-230K

C++ | CUDA | CUDA graph | Cluster scheduling | Compiler

Equity | Health insurance

Mid-level Full Time

San Francisco, Singapore, Amsterdam

1mo 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

1mo 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

1mo ago