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-19 05:16:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
ops
History
Jake VanderPlas
33b989ac9e
refactor: import numpy objects directly in jax.numpy
2025-02-14 12:47:58 -08:00
..
__init__.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
scatter.py
refactor: import numpy objects directly in jax.numpy
2025-02-14 12:47:58 -08:00
special.py
fix grad(logsumexp) to produce 0s where
where
is False
2025-01-08 23:38:06 +00:00
