Save application

featherlessai

Jan 22

Machine Learning Engineer — Training Optimization

Remote Remote · Remote (world)

nodepytorch 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

Join a remote team as an ML Engineer focused on optimizing large-scale model training. Your work will enhance training pipelines and improve model deployment efficiency.

Details

  • Remote work available worldwide
  • Competitive compensation and meaningful equity
  • Strong experience training large neural networks required
  • Experience with PyTorch is required

The work

  • Optimize large-scale model training pipelines for speed and cost
  • Improve distributed training strategies
  • Tune optimizers and batch sizing
  • Build and maintain robust training infrastructure