Principal ML Performance Engineer (GPU Optimization)

Description: Proxima is an AI and data generation company focused on proximity therapeutics and protein interactions. The role involves optimizing ML performance for structural and generative models, including transformers and diffusion models. Requirements: 6+ years of experience in ML systems, HPC, or performance engineering with a relevant degree. Deep knowledge of PyTorch internals and experience with CUDA and Triton. Proficiency in Python and C++, with experience in distributed training at multi-node scale. Ability to set technical direction and mentor engineers. Benefits: Opportunity to work on cutting-edge AI technology in drug discovery. Join a team with a strong track record of partnerships and funding.