Remote Machine Learning Engineer

Summary

Anlatan is seeking a skilled Machine Learning Engineer to join their team in developing cutting-edge AI products. The role involves working with distributed training stacks, contributing to ML codebases, experimenting with new model architectures, and adapting models for their products based on tester feedback.

Requirements

• Experience with pytorch, python
• Past experience with training NLP or Image models
• Able to read cutting edge ML papers, implement them in pytorch and accurately reproduce
• Strong knowledge of computer science fundamentals, including algorithms and data structures
• Machine learning fundamentals
• Good problem-solving and troubleshooting skills

Responsibilities

• Working with our distributed training stack to start training runs and tracking them closely
• Contributing to our ML codebases and adding new models, algorithms and features
• Experimenting with new model architectures and designs, and getting them production ready
• Training and finetuning models, and adapting them to our products by taking tester feedback and iterating over the trained models

Preferred Qualifications

• Knowledge in Kubernetes and SLURM
• Distributed computing knowledge to contribute to our distributed ML training stack
• Experience with JAX
• Experience with Triton

Benefits

• A casual and flexible work environment
• A culture focused on pushing the envelope and having fun while doing it
• Incredibly flexible PTO
• For US Citizens: Health Insurance, Vision and Dental Insurance, Life Insurance and Disability, 401k (3% Match 100%, 2% Match 50%)
• For non-US Citizens: Health Insurance, Vision and Dental Insurance