Expert in Python programming and experience with PyTorch, JAX.
Experience with distributed computing
Experience maintaining an open-source project
Excellent interpersonal skills
Experience and passion to circumvent unexpected roadblocks
Experience in the fast-paced environment of ML research, either through a degree (e.g. PhD) and demonstrated impact on publications, or through relevant product experience.
Experience with CUDA programming, and/or High-Performance Computing is a plus
Education & Experience
M.S. or Ph.D. in Computer Science (or related fields), or B.S. with equivalent experience.