Mixture-of-Depths-Jax. Jax module for the paper: "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"

github.com/dtunai/Mixture-of-Depths-Jax

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.