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

28 results for XLA (Skill/Tech stack)

AI Software Lead – PyTorch & CUDA Runtime (Next-Gen Accelerator) INR 2475K-3380K

C# | C++ | CUDA | Compiler runtime interaction | Compiler/runtime

Senior-level Full Time

Bengaluru, KA, India

2d ago
AI SW Stack Deployment Architect INR 2500K-4500K

API Design | Cloud Computing | Distributed Systems | Edge Computing | Inference Server

Senior-level Full Time

Bengaluru, KA, India

2d ago
AI Performance Optimization Engineer USD 136K-258K

C++ | Continuous batching | Deep learning | Distributed Systems | FSDP

Mid-level Full Time

United States - Remote R

3d ago
AI Performance Optimization Engineer USD 136K-258K

C++ | Cache optimization | Continuous batching | Cutlass | Deep learning

Mid-level Full Time

United States - Remote R

4d ago
AI Performance Optimization Engineer USD 136K-258K

Access patterns | Benchmarking | C++ | Cache optimization | Compiler optimization

Full-time W2 employment | Health benefits | Remote work

Mid-level Full Time

United States - Remote R

4d ago
AI Performance Optimization Engineer USD 159K-264K

C++ | Continuous batching | Cutlass | Deep learning | DeepSpeed

Remote work

Mid-level Full Time

United States - Remote R

4d ago
AI Performance Optimization Engineer USD 136K-258K

Benchmarking | C++ | Compiler optimization | Continuous batching | Debugging

Mid-level Full Time

United States - Remote R

4d ago
AI Performance Optimization Engineer USD 136K-258K

Access Optimization | Attention Optimization | Benchmarking | C++ | Compiler optimization

Mid-level Full Time

United States - Remote R

5d 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

7d ago
Machine Learning Intern/Co-op (Fall, 2026) CAD 60K-60K

CUDA | Distributed Training | GPU | JAX | MLIR

Co-working stipend | Health and dental benefits | Inclusive culture | Lunch stipend | Parental leave top-up

Entry-level Internship

Canada

7d ago
Staff Software Engineer, GPU Performance USD 207K-300K

AMD | CUDA | Code generation | Compiler optimization | Cutlass

Senior-level Full Time

Sunnyvale, CA, USA; Kirkland, WA, USA

7d ago
Research Engineer, Pretraining, DeepMind USD 174K-252K

Fine Tuning | Inference Optimization | JAX | Language Models | Large Language Models

Mid-level Full Time

New York, NY, USA

13d ago
Senior Software Engineer, AI Inference Systems PLN 292K-507K

Algorithms | C++ | CI/CD | CUDA | CUDA Graphs

Hybrid work

Senior-level Full Time

Germany, Remote R

20d ago
Senior AI Compiler Engineer, Algorithms and Code-Generation USD 152K-241K

C# | C++ | CPU architecture | CUDA | Deep learning

Senior-level Full Time

US, CA, Santa Clara, United States

21d ago
Senior Deep Learning Compiler Engineer USD 152K-241K

C# | C++ | CUDA | LLVM | MLIR

Senior-level Full Time

US, CA, Santa Clara, United States

21d ago
Head of AI Embedded Software – NPU Platform

APIs | Compiler infrastructure | Device Drivers | Firmware | Hardware Aware Techniques

Executive-level Full Time

Herzliya, Israel, IL

21d ago
Senior Machine Learning Engineer, Runtime and Serving USD 213K-263K

Benchmarking | Buffer management | C++ | CUDA | Concurrent Systems

Senior-level Full Time

Mountain View, CA, USA

21d ago
Senior Deep Learning Compiler Verification Engineer USD 140K-224K

C++ | Formal verification | Graph optimization | IR lowering | JAX

Comprehensive benefits package | Equity

Senior-level Full Time

US, CA, Santa Clara, United States

22d ago
SOC Architect, XProf USD 147K-211K

C# | C++ | Compiler profiling | Data Analysis | Data Visualization

Senior-level Full Time

Sunnyvale, CA, USA

22d ago
Machine Learning Engineer, Runtime & Optimization USD 213K-263K

C++ | CUDA | Deep learning | JAX | Machine Learning

Company benefits program | Discretionary annual bonus | Equity incentive plan

Senior-level Full Time

Mountain View, California, USA

27d ago
ML Runtime Optimization Engineer USD 159K-199K

CPU | CUDA | Deep learning | Embedded Systems | GPU

401k match | Dental insurance | Disability insurance | Health insurance | Learning stipend

Senior-level Full Time

Sunnyvale, California, United States

27d ago
Manager, Deep Learning – Autonomous Vehicles and Robotics USD 224K-431K

Agentic AI | Autonomous Driving | Compiler technology | Curriculum Design | Deep learning

Mid-level Full Time

US, CA, Santa Clara, United States

29d ago
Senior Software Engineer, Kernels and Performance, Core ML Frameworks USD 174K-252K

Autotuning | Benchmarking | C++ | CUDA | Code generation

Senior-level Full Time

Sunnyvale, CA, USA

1mo ago
Principal Machine Learning Engineer, Mobile AI Inference Optimization USD 278K-347K

ARM Mali | Android | Apple Neural Engine | C++ | CoreML

Commute subsidy | Employee assistance program | Employee resource groups | Employee stock ownership | Generous vacation and personal days

Senior-level Full Time

Mountain View, CA, USA

1mo ago
Senior Software Engineer, Machine Learning, Core ML USD 174K-252K

C++ | Compiler optimization | Data Processing | Data parallelism | Debugging

Senior-level Full Time

Mountain View, CA, USA

1mo ago
Principal Deep Learning Communication Architect USD 272K-431K

3D Parallelism | CUDA | Context Parallelism | Data parallelism | DeepSpeed

Senior-level Full Time

US, CA, Santa Clara, United States

1mo ago
Senior Software Engineer, LLM Performance USD 180K-339K

C++ | CUDA | Cutlass | FlashAttention | FlashInfer

Senior-level Full Time

SF Bay Area (Hybrid) R

1mo ago
Compiler Engineer - Machine Learning Compiler USD 170K-251K

Backend Development | C++ | Code generation | Compiler Intermediate Representation | Constraint Programming

Mid-level Full Time

Palo Alto, CA

1mo ago