Research Engineer, Ml Infrastructure

Research Engineer - ML Infrastructure at Chai Discovery. This role focuses on making ML models performant, resource-efficient, and reliable at scale by developing core frameworks for model training and evaluation. Responsibilities include building the training stack, optimizing workloads, profiling training runs, and ensuring scalability and reliability of ML training jobs.

Skills and focus areas

• ml infrastructure
• model training
• model evaluation
• parallelism
• quantization
• custom kernels
• performance profiling
• bottleneck elimination
• throughput monitoring
• utilization monitoring
• uptime monitoring
• fault tolerance