Senior ML Engineer (Token Factory)

📋 Description

• Build inference and fine-tuning tech for foundation models at scale.
• Improve model quality and training efficiency for large ML workloads.
• Address bottlenecks in LLM inference for production performance.
• Develop training/evaluation pipelines using JAX (e.g., speculative decoding).
• Experiment with dense and mixture-of-experts models and autoregressive/parallel approaches.
• Translate research into robust, production-quality ML systems.

🎯 Requirements

• Deep understanding of ML/reinforcement learning foundations.
• Strong expertise in modern DL for language processing/generation.
• Experience training large models across multiple nodes.
• Performance optimization for large NN training; sharding, custom kernels, hardware-aware.
• Strong Python software engineering skills.
• Extensive experience with JAX.

🎁 Benefits

• Competitive compensation.
• Career growth and continuous learning opportunities.
• Flexible working environment with high ownership.
• Impactful, large-scale AI/machine learning projects.
• Collaborative culture with experienced engineers/researchers.
• International environment with diverse, highly skilled teams.