Senior ML Engineer (Token Factory)

📋 Description

• Develop and improve fine-tuning methods for foundation models (LoRA-based, full-parameter).
• Optimize model quality and training efficiency on large ML workloads.
• Address bottlenecks in LLM inference for production performance.
• Build training/evaluation pipelines using JAX (speculative decoding, inference opt).
• Experiment with dense and mixture-of-experts architectures.
• Develop scaling laws for model development and resources.

🎯 Requirements

• Deep ML theory and reinforcement learning foundations.
• Expertise in modern DL for language processing/generation.
• Experience training large models across multi-node clusters.
• Strong optimization for large NN training, including sharding & hardware.
• Strong Python software engineering skills.
• Extensive experience with JAX and modern ML frameworks.

🎁 Benefits

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