-
flax Public
Forked from google/flaxFlax is a neural network library for JAX that is designed for flexibility.
Python Apache License 2.0 UpdatedFeb 16, 2023 -
-
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python Apache License 2.0 UpdatedJul 13, 2022 -
-
-
-
-
-
-
-
-
-
-
-
-
{{ message }}