"kernel fusion" Jobs

13 open tech roles matching “kernel fusion”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: CUDA, JAX, Python. Every listing is re-checked daily and closed roles are removed.

Showing 13 of 13 results

Staff Software Engineer - GenAI Performance and Kernel

Databricks San Francisco, California $190.9k–$232.8k/yr Published 11 months ago

Triton oneDNN CUTLASS CUDA NVProf

Apply →

Senior AI Infrastructure Engineer - Model Training Visa

Join Kodiak Robotics as a Senior AI Infrastructure Engineer to optimize model training for autonomous technology.

Kodiak Robotics Mountain View, CA $190k–$260k/yr Published 2 months ago

Megatron CUDA DeepSpeed PyTorch Python

Flexible on stack

Apply →

Member of Technical Staff, TPU Performance Engineering Visa

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact Singapore S$200k–S$400k/yr Published 2 months ago

JAX vLLM ML Pallas TPU

Flexible on stack

Apply →

Inference Engineering, Co-op

Join Inferact as a co-op student to work on cutting-edge AI inference systems in a hands-on engineering role.

Inferact San Francisco, California, United States Published 1 day ago

C++ Triton JAX GPU programming CUDA

Flexible on stack

Apply →

Member of Technical Staff, TPU Performance Engineering Hybrid Visa

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact San Francisco $200k–$400k/yr Published 2 months ago

JAX vLLM ML Pallas TPU

Flexible on stack

Apply →

GPU Performance Engineer

Join Genmo as a GPU Performance Engineer to optimize video generation models and achieve significant performance improvements.

Genmo San Francisco HQ Published 1 year ago

C++ Nsight Systems Triton CUDA NVProf

Flexible on stack

Apply →

AI Performance Engineer

Join Applied Intuition as an AI Performance Engineer to optimize large-scale machine learning workloads in a collaborative environment.

Applied Intuition Sunnyvale Published 1 month ago

C++ Ray ONNX Runtime Megatron CUDA

Flexible on stack

Apply →

Performance Engineer (Inference, Training & GPU)

Join World Labs as a Performance Engineer to optimize AI models for speed and efficiency in a cutting-edge research environment.

World Labs San Francisco $200k–$300k/yr Published 4 months ago

FP8 C++ Triton JAX CUDA

Flexible on stack 70% coding

Apply →

Performance Engineer, GPU Hybrid Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 11 months ago

FP8 NVLink Triton JAX CUTLASS

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research with a focus on hardware-aware algorithms and impactful implementations.

Graphcore Bristol, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on innovative AI hardware solutions.

Graphcore London, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on next-gen AI hardware.

Graphcore Cambridge, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

Research Engineer

Join Mind Robotics as a Research Engineer to build scalable systems for training large ML models in a hands-on robotics environment.

Mind Robotics Palo Alto Published 7 months ago

JAX PyTorch Python

Flexible on stack 70% coding

Apply →