mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src
History
Peter Hawkins 10b7d7d7c2 Move implementation of jax.lax into jax._src.lax. 
Remove lax_ prefixes from jax/_src/lax filenames, since they aren't needed any longer to avoid name conflicts.
2020-10-17 16:09:21 -04:00
..
image
Move implementation of jax.image to jax._src.image.
2020-10-17 11:37:55 -04:00
lax
Move implementation of jax.lax into jax._src.lax.
2020-10-17 16:09:21 -04:00
nn
Move jax.nn implementation into jax._src.nn.
2020-10-17 13:45:01 -04:00
numpy
Move implementation of jax.lax into jax._src.lax.
2020-10-17 16:09:21 -04:00
ops
Move jax.ops implementation into jax._src.ops.
2020-10-17 11:45:28 -04:00
scipy
Move jax.numpy internals into jax._src.numpy.
2020-10-16 20:35:19 -04:00
__init__.py
Move most of the implementation of jax.scipy into jax._src.scipy.
2020-10-16 17:04:25 -04:00