Mixture of Experts (MoE) Jobs - February 2026

Sort by

Search

Location

Has Compensation Summary

Has Contact Info

Remote

Onsite

Tech Stack

Title

Reset

AMD

Posted 2 days, 18 hours ago

Building Wave, a high-performance GPU programming language and compiler for machine learning workloads. Own end-to-end performance of Wave's kernels. Work includes hand-tuned GPU kernels (GEMM, Attention, MoE, decoding); instruction-level optimizati…

Roles
GPU Performance Software Development Engineer

Tech Stack
CUDA GPU intrinsics MMA rocprof Attention HIP GEMM GPU compiler pipelines MFMA profiling tools Mixture of Experts (MoE) NVIDIA Nsight Python-embedded DSL MLIR dialects inline assembly MLIR

Locations

San Jose, CA