Find jobs in AI/ML, Data Science and Big Data
21 results
for XLA
(Skill/Tech stack)
-
Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | FlaxSenior-level Full TimeMountain View, California, United States, New …19h ago
-
Senior-level Full TimeChina, Shanghai1d ago
-
Entry-level Full TimeHsinchu City, Hsinchu City, Taiwan5d ago
-
C++ | Co-design | Compilers | Data Analysis | DebuggingSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA10d ago
-
Software Engineer, Systems ML USD 141K-208KC plus plus | CUDA | Co-design | Compiler optimization | Deep learningSenior-level Full TimeBellevue, WA | Menlo Park, CA …12d ago
-
Benchmarking | C++ | Continuous batching | Deep learning | Disaggregated servingSenior-level Full TimeSunnyvale, CA, USA13d ago
-
Research Engineer, Multi Agent Learning, DeepMind GBP 100K-200KAgent systems | Algorithmic game theory | Autograd | C++ | Computational economicsMid-level Full TimeLondon, UK18d ago
-
Research Engineer, Pretraining, DeepMind USD 174K-253KFine Tuning | Inference Optimization | JAX | Language Models | Large Language ModelsMid-level Full TimeNew York, NY, USA20d ago
-
C++ | CUDA | Compute Graphs | Distributed Systems | GPUCross-functional collaboration | Engineering autonomy | Flexible working modelMid-level Full TimeGdansk, Poland27d ago
-
Staff Compiler Engineer - PyTorch + Kernel DSLPLATE USD 163K-253KAutotuning | Collective Primitives | Cost Based Compilation | Custom ISA | Cutlass401k | Adoption support stipend | Charitable giving match | Fertility care stipend | Flexible work environmentSenior-level Full TimeSan Jose, California, United States27d ago
-
C# | C++ | Deep learning | Domain-specific language | LLVMSenior-level Full TimeChina, Shanghai28d ago
-
Entry-level Full TimeNew York, NY, United States29d ago
-
Entry-level Full TimeNew York, NY, United States29d ago
-
AI Computing Architect CNY 240K-480KArchitecture simulation | C# | C++ | CUDA | Computer ArchitectureSenior-level Full TimeChina, Shanghai1mo ago
-
Machine Learning Engineer USD 145K-180KAWS Batch | AWS EC2 | AWS Lambda | Airflow | Async ProcessingCompany paid life insurance | Medical, dental, and vision coverage | Mental well-being resources | Mentorship and resources | Paid parental leaveSenior-level Full TimeNew York, NY, US, 102811mo ago
-
Research Scientist (Visual Generative AI & World Models) GBP 195K-270KATen | C++ | CUDA | Calculus | Deep learningDental plan | Employee assistance programme | Employee wellbeing support | Flexible working | Generous annual leaveSenior-level Full TimeLondon, UK1mo ago
-
AI Software Lead – PyTorch & CUDA Runtime (Next-Gen Accelerator) INR 2475K-3380KC# | C++ | CUDA | Compiler runtime interaction | Compiler/runtimeSenior-level Full TimeBengaluru, KA, India1mo ago
-
AI SW Stack Deployment Architect INR 2500K-4500KAPI Design | Cloud Computing | Distributed Systems | Edge Computing | Inference ServerSenior-level Full TimeBengaluru, KA, India1mo ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | CUDA kernel | CUDA kernel optimization | Computer ArchitectureEquity | Health benefits | Paid time offSenior-level Full TimeUS, CA, Santa Clara, United States1mo ago
-
Machine Learning Intern/Co-op (Fall, 2026) CAD 60K-60KCUDA | Distributed Training | GPU | JAX | MLIRCo-working stipend | Health and dental benefits | Inclusive culture | Lunch stipend | Parental leave top-upEntry-level InternshipCanada1mo ago
-
Member of Technical Staff, Performance Optimization USD 175K-220KCUDA | CUPTI | Distributed Systems | GPU Profiling | InfinibandSenior-level Full TimeSan Mateo, CA1mo ago