This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
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
Jake Vanderplas
d8f799391b
COPYBARA_INTEGRATE_REVIEW=
https://github.com/google/jax/pull/17027
from jakevdp:dtypes-annotations a116a9c498a7b085f9b3fec93b37da12289f6e31
...
PiperOrigin-RevId: 554905739
2023-08-08 20:38:44 +00: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
[shape_poly] Cleanups for the shape polymorphism APIs.
2023-06-30 15:56:57 +03:00
initializers.py
COPYBARA_INTEGRATE_REVIEW=
https://github.com/google/jax/pull/17027
from jakevdp:dtypes-annotations a116a9c498a7b085f9b3fec93b37da12289f6e31
2023-08-08 20:38:44 +00:00
