mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/nn
History
Marcus Chiam 45c2f31887 Added shape error checking for compute_fans 
Update tests/nn_test.py

Co-authored-by: Jake Vanderplas <jakevdp@google.com>
2023-01-18 20:59:11 -08:00
..
__init__.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
functions.py
Use jax.Array by default for doctests
2022-11-15 11:52:22 -08:00
initializers.py
Added shape error checking for compute_fans
2023-01-18 20:59:11 -08:00