Description
Anlatan is a company working on cutting edge AI and making the best possible AI-fueled products. We pride ourselves on being user-friendly, privacy-safe, and censorship-free. We're a ragtag team of personalities, focused on crafting the best of the best in the AI world.
We are seeking a skilled Machine Learning Engineer to join our team. In Anlatan, you will be developing code and model architectures to train cutting edge LLMs and Image models in our compute cluster.
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.
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.
Bonus Points
- Knowledge in Kubernetes and SLURM.
- Distributed computing knowledge to contribute to our distributed ML training stack.
- Experience with JAX.
- Experience with Triton.