11mo
Not stated
Berlin
Hybrid
Full-time
MAMirelo AI
Experience in implementing custom GPU kernels and integrating them into PyTorch. Experience with diffusion and autoregressive models and understanding of their specific optimization challenges Familiarity with high-performance …
PyTorch, Go