Machine Learning Engineer

Job Summary

Joining a dynamic team, the full-time remote Machine Learning Engineer will develop code and model architectures for cutting-edge LLMs and image models while working with distributed training stacks and contributing to ML codebases.

Key responsibilities

• Working with the distributed training stack to initiate and monitor training runs
• Contributing to ML codebases by adding new models, algorithms, and features
• Experimenting with new model architectures and preparing them for production based on tester feedback

Required qualifications

• Experience with PyTorch and Python
• Past experience training NLP or image models
• Able to read and implement cutting-edge ML papers in PyTorch
• Strong knowledge of computer science fundamentals, including algorithms and data structures
• Understanding of machine learning fundamentals