Training Infrastructure Engineer

Remote Remote · Berlin

gopytorch researchfulltime

ashby

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

You will develop and optimize the training infrastructure for generative AI models in Berlin. This role involves working with GPU behavior, training pipelines, and scalable systems for model training.

Details

Berlin, remote work available
Competitive salary and equity
Experience with Go and PyTorch required

The work

Profile, debug, and optimize GPU operations
Analyze and improve the training pipeline
Set up scalable systems for experiment tracking
Design and maintain large-scale ML training clusters