Research Engineer, Large-Scale Training

Onsite San Francisco

pythonnodepytorch research

greenhouse

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 optimize large-scale training infrastructure and integrate new model architectures. Your work will enhance the fine-tuning experience for customers at Together AI in San Francisco.

Details

San Francisco, work from office
$200,000 - $290,000
Strong programming skills in Python and PyTorch
Experience training or fine-tuning large neural networks in multi-GPU or multi-node environments

The work

Design, implement, and optimize core components of large-scale training infrastructure
Integrate new model architectures and optimize performance for production fine-tuning workloads
Profile distributed training workloads to identify and eliminate bottlenecks
Build and maintain experimental infrastructure that accelerates research