Training: ML Framework Engineer

Job summary

Responsibilities

As a Training: ML Framework Engineer, you will work on improving the training throughput for our internal training framework, while enabling researchers to experiment with new ideas.

• Apply the latest techniques in our internal training framework to achieve impressive hardware efficiency for our training runs
• Profile and optimize our training framework
• Work with researchers to enable them to develop the next generation of models
• Have run small scale ML experiments
• Love figuring out how systems work and continuously come up with ideas for how to make them faster while minimizing complexity and maintenance burden

Tech stack

• Python