Loreon
Labs
Platform
Docs
Home
Ecosystems
Python
modelling_in_jax
Building scalable models in pure JAX
Python
Emerging
GitHub
Stars
2
Forks
—
Contributors
1
Last push
8mo ago
Recent commits
Latest commits.
add doc for kernel init
801db69
AakashKumarNain
8mo ago
format code
15c4fbd
AakashKumarNain
8mo ago
add MHA and rename GQA
bc8c322
AakashKumarNain
8mo ago
fix formatting
a3f63de
AakashKumarNain
9mo ago
update pre-commit hooks
62a39a9
AakashKumarNain
9mo ago
remove redundant parts
17f1ff8
AakashKumarNain
9mo ago
precommit test
3037bcd
AakashKumarNain
9mo ago
format code
5852d94
AakashKumarNain
9mo ago
Top contributors
Builders behind this project.
AakashKumarNain
45 commits
