Generative AI - ML System Engineering

Experience in machine learning or high performance graphics Solid practical understanding of at least one machine learning framework (e.g. PyTorch, JAX) Strong ability to write beautiful and maintainable code in Python and/or C++ Ability to learn fast and dive into new concepts or complex codebases Performance and efficiency oriented mindset Strong communication skills for working in a globally distributed team Experience with PyTorch internals (torch.compile, fully_shard) Experience building Triton kernels Experience with large-scale distributed training and parallelization techniques Experience with diffusion models in 3D or video Experience with low precision bf16 or fp8 training

Build and debug on top of modern PyTorch Build clean and intuitive training infrastructure Optimize for high throughput & efficient distributed model training Implement and maintain 3D specific custom operators in Triton or CUDA Implement and maintain novel data-loading framework and libraries Build efficient inference endpoints with complex multi-stage model pipelines Optimize models through compilation, fusion, quantization, etc.