Magic · 1 year ago
GPU Kernel Engineer
San Francisco
Full-time
Hybrid
Mid LevelMagic is an AI coding startup that enables developers to work with AI to find code for building apps. As a GPU Kernel Engineer, you will design efficient implementations of novel model architectures and optimize kernels to ensure high throughput and low latency during training and inference.
Artificial Intelligence (AI)Information TechnologyMachine Learning
No H1B
Responsibilities
Write efficient custom kernels for training and inference in CUDA/CuTe/Cutlass
Optimize inference for our novel architectures, both by writing more efficient code and thinking about how we can sacrifice accuracy for speed
Understand and optimize for H100 GPUs
Think beyond the kernel level to the broader scheme of how we train these models and suggest improvements
Qualification
Required
Understands and has worked on GPU programming, ideally matmul-heavy workloads
Benefits
401K with 6% match
Flexible working hours
Company
Magic
Magic is an AI coding startup that enables developers to work with AI to find code for building apps.
Founded in 2022
2-10 employees
Funding
Current Stage
Growth StageTotal Funding
$465.93MKey Investors
Flat CapitalNFDG VenturesCapitalG
2025-01-14Series Unknown· $0.81M
2024-08-29Series Unknown· $320M
2024-02-16Series B· $117M
Recent News
2025-11-13
TechWire Asia
2025-09-12
2024-11-01
Company data provided by crunchbase