richard
/
learn_jax
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
learn_jax/equinox
History
Richard Wong 0762c02b31 Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00
..
.gitignore Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00
handling_state_equinox.py Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00
handling_state_flax.py Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00
mnist.py Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00
t5_simple_train_model.py Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00
t5_train_model.py Feat: implemented attention layer in equinox 2024-10-06 23:52:42 +09:00