Requirements

A current undergraduate, master's or PhD student in machine learning or a related discipline
Strong object-oriented programming skills and experience working with Python, PyTorch and NumPy are desirable
Experience in one or more advanced optimization methods, modern ML techniques, HPC, profiling, model inference; you don't need to have all of the above
Excellent machine learning reasoning skills, with the ability to develop your own models when standard approaches are insufficient
Strong communication and collaboration skills, with the ability to work effectively in a team with complementary expertise

What You'll Be Doing

Implementing ideas from published research papers
Writing custom libraries to efficiently train models on petabytes of data
Reducing model training times by hand optimizing machine learning operations
Profiling custom machine learning architectures to identify performance bottlenecks
Evaluating the latest hardware and software in the machine learning ecosystem