https://github.com/hmunachi/nanodl
A Jax-based library for designing and training transformer models from scratch.
attention
attention-mechanism
deep-learning
distributed-training
flax
gpt
jax
llama
machine-learning
mistral
nlp
transformer
Added: about 1 year ago - Last Synced: 11 months ago
- Created: August 22, 2023
