Senior ML Engineer (Token Factory)

📋 Description

• Develop and optimize fine-tuning methods for foundation models.
• Improve model quality and training efficiency at scale.
• Address bottlenecks in LLM inference for production use.
• Build JAX-based training/evaluation pipelines (e.g., speculative decoding).
• Experiment with dense and mixture-of-experts models.
• Translate research into robust production ML systems.

🎯 Requirements

• Deep ML/ RL theory and foundations.
• Expertise in deep learning for NLP.
• Experience training large models across nodes.
• Strong performance optimization for large NN training.
• Proficiency in Python.
• Extensive experience with JAX.

🎁 Benefits

• Competitive compensation.
• Career growth and continuous learning.
• Flexible environment with ownership.
• Impactful, large-scale AI projects.
• Collaborative culture with experienced teams.
• International environment with diverse teams.