learn_jax/equinox
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