Find your next role
Discover roles across ambitious teams
4,495 jobs
- More jobs
Egen
ML Platform Engineer, Distributed Training
Remote · Full time · Senior level
$159k–$219k / year
About the role
Lead the full lifecycle of AI features focusing on distributed training and scalable production services. Collaborate with clients to build high-performance AI architectures addressing real-world constraints.
What you’ll do
• Work with client infrastructure on GKE training platform. • Diagnose and fix platform failures at root cause. • Tune multi-node GPU networking for training throughput. • Run and debug multi-node distributed training with PyTorch and Ray Train. • Design and run performance benchmarks and model evaluations.
What you’ll bring
• 5+ years engineering experience with GPU training infrastructure. • Strong Python software engineering skills. • Experience with GKE and GPU scheduling on Google Cloud. • Proficient in distributed PyTorch and Ray Train multi-node jobs. • Knowledge of RDMA, GPUDirect, NCCL tuning and benchmarking.
Life at Egen
Competitive salary with comprehensive health insurance, paid leave, 401(k) match, and employee referral bonuses. Inclusive and supportive workplace with accommodations available.
About Egen
Egen is a fast-growing company focused on data-driven engineering solutions using advanced platforms like Google Cloud and Salesforce to drive impactful client outcomes.