mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax
History
Yash Katariya dbeb97d394 Create 0.2.21 jax release 
PiperOrigin-RevId: 398528427
2021-09-23 11:00:31 -07:00
..
_src
Merge pull request #7983 from jakevdp:fix-std-basis
2021-09-23 09:15:51 -07:00
experimental
Merge pull request #7993 from gnecula:tf_share_const
2021-09-23 09:23:06 -07:00
image
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
interpreters
Move contents of jax.lib to jax._src.lib.
2021-09-23 06:33:55 -07:00
lax
Remove export of jax.lax.partial.
2021-09-14 16:17:50 -04:00
lib
Move contents of jax.lib to jax._src.lib.
2021-09-23 06:33:55 -07:00
nn
Fix package exports
2021-09-14 13:55:55 -07:00
numpy
jax.numpy: add wrapper for np.insert
2021-09-16 12:18:32 -07:00
ops
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
scipy
Fix package exports
2021-09-14 13:55:55 -07:00
tools
Move contents of jax.lib to jax._src.lib.
2021-09-23 06:33:55 -07:00
__init__.py
Move contents of jax.lib to jax._src.lib.
2021-09-23 06:33:55 -07:00
abstract_arrays.py
Move jax.abstract_arrays to jax._src.abstract_arrays.
2021-06-02 06:25:22 -07:00
api_util.py
Move contents of jax.api_util to jax._src.api_util and add a forwarding shim.
2021-09-08 09:00:56 -07:00
cloud_tpu_init.py
Automatically initialize Cloud TPU topology env vars if running on a Cloud TPU VM.
2021-03-10 09:15:31 -08:00
config.py
Move jax.config to jax._src.config.
2021-04-19 08:53:12 -07:00
core.py
Switch internal users of jax.util.partial to use functools.partial.
2021-09-13 21:09:58 -04:00
custom_derivatives.py
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
dlpack.py
[call_tf] Add support for DLPack to avoid copying arrays in eager mode
2021-02-13 12:49:51 +02:00
dtypes.py
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
errors.py
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
flatten_util.py
Move implementation of jax.flatten_util to jax._src.flatten_util. Add a jax.flatten_util shim.
2021-09-08 13:54:25 -07:00
jaxpr_util.py
[JAX] Move jax.util to jax._src_util.
2021-01-11 14:21:07 -08:00
linear_util.py
Cleanup: switch to new version of super()
2021-08-05 13:11:07 -07:00
prng.py
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
profiler.py
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
py.typed
Improve type annotations
2021-01-13 10:26:35 -05:00
random.py
Merge pull request #7847 from google:prng-type
2021-09-14 12:55:28 -07:00
test_util.py
Move contents of jax.lib to jax._src.lib.
2021-09-23 06:33:55 -07:00
tree_util.py
Add PEP484-compatible export for jax and its subpackages
2021-09-13 14:08:48 -07:00
util.py
Remove jax.util.partial.
2021-09-20 20:32:49 -04:00
version.py
Create 0.2.21 jax release
2021-09-23 11:00:31 -07:00