Senior ML Engineer (Token Factory)

📋 Description

• Lead and improve fine-tuning methods for foundation models
• Optimize training efficiency on large-scale ML workloads
• Address bottlenecks in LLM inference for production
• Build training/evaluation pipelines using JAX
• Experiment with dense and mixture-of-experts models
• Develop scaling laws for model and resource planning

🎯 Requirements

• Deep ML theory, RL foundations
• Expertise in modern DL for language
• Experience training large models across nodes
• Performance optimization for large NN, sharding, kernels
• Strong Python software engineering
• Experience with JAX

🎁 Benefits

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