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

22 results for Kernel Fusion (Skill/Tech stack)

Machine Learning Backend Engineer Graduate (AML MLDev) - 2027 Start USD 128K-256K

C++ | CUDA | CUDA toolkit | Code generation | Computation Scheduling

Entry-level Full Time

San Jose, California, United States

4d ago
Machine Learning Backend Engineer Graduate (AML MLdev) - 2027 Start SGD 81K-100K

Bottleneck diagnosis | Code generation | Computation Scheduling | Distributed Systems | Graph optimization

Entry-level Full Time

Singapore, Singapore

6d ago
Sr. Machine Learning Engineer USD 200K-260K

AWS | Apache Airflow | Asynchronous processing | Azure | CI/CD

Senior-level Full Time

United States and Canada, CA, USA

7d ago
Research Engineer USD 200K-300K

Attention Optimization | Data parallelism | Distributed Training | Experiment tracking | GPU Computing

Mid-level Full Time

Palo Alto

15d ago
Staff Engineer, Inference Optimizations USD 191K-239K

AMD GPU | Artificial Intelligence | Attention Mechanisms | Bandwidth Optimization | CUDA

Conference reimbursement | Education reimbursement | Employee assistance program | Flexible time off | LinkedIn Learning access

Senior-level Full Time

Seattle

17d ago
Staff Engineer, Inference Optimizations USD 191K-239K

BF16 | Bandwidth Optimization | CUDA | Expert Routing | FP4

Conference reimbursement | Education reimbursement | Employee assistance program | Flexible time off | LinkedIn Learning access

Senior-level Full Time

Denver R

17d ago
Staff Engineer, Inference Optimizations USD 191K-239K

Attention Mechanisms | BF16 | CUDA | CUDA kernels | Distributed Systems

Conference reimbursement | Education reimbursement | Employee assistance program | Flexible time off | LinkedIn Learning

Senior-level Full Time

Boston R

17d ago
Staff Engineer, Inference Optimizations USD 191K-239K

AMD GPU | Attention Optimization | Bandwidth Optimization | Benchmarking | CUDA

Conference reimbursement | Employee assistance program | Flexible time off | LinkedIn Learning access | Local Employee Meetups

Senior-level Full Time

Austin R

17d ago
Staff Engineer, Inference Optimizations USD 191K-239K

Attention Optimization | BF16 | CUDA | FP4 | FP8

Conference reimbursement | Education reimbursement | Employee assistance program | Employee stock purchase program | Equity compensation

Senior-level Full Time

San Francisco R

17d ago
AI Researcher — Inference Optimization USD 184K-293K

CUDA | Deep learning | Distributed Systems | Dynamic batching | GPU Inference

Mid-level Full Time

Remote (world) R

20d ago
Research Engineer, Algorithms USD 300K-400K

Analog computation | Attention Mechanism | Efficient Attention | Inference Optimization | KV cache

Mid-level Full Time

New York City

22d ago
ML Compiler Engineer, TPU Performance Optimizations INR 2500K-5000K

C++ | CUDA | Compiler optimization | Computer Architecture | Debugging

Senior-level Full Time

Bengaluru, Karnataka, India

26d ago
AI Research Engineer GBP 100K-200K

ATen | Asynchronous execution | C++ | CUDA | Compilation

Annual leave | Dental plan | Employee assistance programme | Flexible working | Health cash plan

Mid-level Full Time

Cambridge, UK

27d ago
AI Research Engineer GBP 100K-200K

ATen | Asynchronous execution | C++ | CUDA | Compilation

Annual leave | Dental plan | Employee assistance programme | Flexible working | Health cash plan

Mid-level Full Time

Bristol, UK

27d ago
AI Research Engineer GBP 100K-200K

ATen | Asynchronous execution | C++ | CUDA | Compilation

Annual leave | Dental plan | Employee assistance program | Flexible working | Health cash plan

Mid-level Full Time

London, UK

27d ago
Research Scientist (Visual Generative AI & World Models) GBP 200K-250K

ATen | Architecture Design | Asynchronous execution | C++ | CUDA

Annual leave | Dental plan | Employee assistance program | Flexible working | Generous parental leave

Senior-level Full Time

London, UK

27d ago
Research Scientist (Visual Generative AI & World Models) GBP 146K-250K

ATen | Asynchronous execution | C++ | CUDA | Calculus

Dental plan | Employee assistance programme | Flexible working | Generous annual leave | Generous parental leave

Senior-level Full Time

Bristol, UK

27d ago
Research Scientist (Visual Generative AI & World Models) GBP 146K-250K

ATen | Asynchronous execution | C++ | CUDA | Calculus

Dental plan | Employee assistance programme | Equal opportunity support | Flexible working | Generous annual leave

Senior-level Full Time

Cambridge, UK

27d ago
Machine Learning Engineer - Model Inference USD 166K-230K

C++ | Cloud infrastructure | Containers | Continuous batching | Distributed Systems

Mid-level Full Time

Cupertino

1mo ago
Research Scientist - Distributed Machine Learning USD 180K-287K

BF16 | CUDA | CUDA kernels | DeepSpeed | Distributed Training

401k | Dental insurance | Disability insurance | Employee assistance program | Health insurance

Mid-level Full Time

Sunnyvale, CA

1mo ago
Machine Learning Infrastructure Engineer USD 216K-330K

CUDA | DeepSpeed | Distributed Systems | Distributed Training | FSDP

Mid-level Full Time

Sunnyvale, CA

1mo ago
Sr. Staff Machine Learning Researcher - Model Training & Optimization CAD 100K-500K

Deep learning | Distributed Training | Flash Attention | Inference Optimization | Kernel Fusion

Hybrid work

Senior-level Full Time

Toronto, Ontario, Canada

1mo ago