The evaluation notebook for the NeurIPS paper "Sparse is Enough in Scaling Transformers".
The evaluation notebook for the NeurIPS paper "Sparse is Enough in Scaling Transformers".

PiperOrigin-RevId: 405537505

google/trax,google/trax