ML Infrastructure Engineer – Large-Scale Training (Relocation)

A tech-focused research team is searching for a Machine Learning Engineer to develop the infrastructure for large-scale training and experimentation of neural networks. The ideal candidate will design and maintain end-to-end ML pipelines, optimize performance for transformer models, and collaborate with researchers. Strong experience in deep learning and high proficiency in PyTorch or JAX is required. The position offers salary ranging from $150,000 to $400,000 per year, with opportunities for relocation assistance. This role is in-person in San Francisco. #J-18808-Ljbffr