Job Details

Machine Learning Engineer

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

We're working fast to send ripples through the ever-moving field of AI, we offer a competitive salary package, and of course, we're all pretty fun to work with.

We offer a competitive salary and benefits, including:

• A casual and flexible work environment
• A culture focused on pushing the envelope and having fun while doing it
• Incredibly flexible PTO
• 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

Beyond all else, our members are encouraged to:

• Explore and Experiment
• Work Stress-free
• Ask For Help
• Interact with the Community