Research
Kolibri
I contributed to Kolibri pre-training through expert load balancing, training optimization and hyperparameter tuning. Kolibri is an English-German MoE language model with 78.1B total parameters and 3.46B active parameters per token.
Technical report (opens in a new tab) · Model (opens in a new tab)
Exact Quantile Balancing and Load-Error Injection for Mixture-of-Experts
I co-developed EQB and LEI for global and local expert load balancing in MoE models.
SOMBRERO: Measuring and Steering Boundary Placement in End-to-End Hierarchical Sequence Models
We measure and steer learned segment boundaries in byte-level language models, aligning computation with positions that are harder to predict.
©LI JiazeRSS