HMUNACHI / nanodl

A Jax-based library for designing and training transformer models from scratch.
MIT License
271 stars 11 forks source link