vision_transformer by google-research

None

updated at May 12, 2024, 11:40 a.m.

95 -1

9,367 +45

1,206 +3

dm-haiku by deepmind

JAX-based neural network library

updated at May 12, 2024, 12:33 p.m.

39 +0

2,816 +10

226 +0

equinox by patrick-kidger

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

updated at May 12, 2024, 12:38 p.m.

21 +0

1,834 +9

127 +0

chex by deepmind

None

updated at May 12, 2024, 12:40 p.m.

17 +0

720 +4

40 +0