Senior ML Engineer (Token Factory)

📋 Description

• Develop and improve fine-tuning methodologies for foundation models (LoRA-based and full-parameter
• Optimize model quality and training efficiency for large-scale ML workloads.
• Address bottlenecks in LLM inference to improve production performance and resource use.
• Build training/evaluation pipelines using JAX for speculative decoding and advanced inference
• Experiment with architectures: dense, mixture-of-experts; autoregressive and parallel approaches.
• Translate research into robust, scalable production ML systems.

🎯 Requirements

• Deep understanding of ML theory and reinforcement learning fundamentals.
• Expertise in modern DL techniques for language processing/generation.
• Experience training large models across multiple compute nodes.
• Strong performance optimization for large NN training (sharding, kernels, hardware).
• Strong Python software engineering skills.
• Extensive experience with JAX and modern DL frameworks.

🎁 Benefits

• Competitive compensation.
• Career growth and continuous learning opportunities.
• Flexible working environment with ownership.
• Opportunity to work on impactful, large-scale AI projects.
• Collaborative culture with experienced engineers and researchers.
• International environment with diverse teams.