GitHub repository: HMUNACHI/nanodl
Primary language: Python
Description: A Jax-based library for designing and training transformer models from scratch.