mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 06:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax
History
George Necula 0bc70bbd73 Disable jax2tf test recently added in cl/688976685. 
See failure: https://github.com/jax-ml/jax/actions/runs/11514933009/job/32054580529?pr=24183

PiperOrigin-RevId: 689703645
2024-10-25 02:12:34 -07:00
..
_src
Merge pull request #24511 from mattjj:improve-concreteness-error-in-remat
2024-10-24 13:05:36 -07:00
example_libraries
update example optimizers library docstring
2024-09-03 23:40:47 -07:00
experimental
Disable jax2tf test recently added in cl/688976685.
2024-10-25 02:12:34 -07:00
extend
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
image
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
interpreters
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
lax
Activate the FFI implementation of SVD on GPU.
2024-10-17 17:57:06 -07:00
lib
Deprecate a number of APIs in jax.lib.xla_client.
2024-10-11 11:42:40 -07:00
nn
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
numpy
Make jnp.subtract a ufunc
2024-10-21 10:11:51 -07:00
ops
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
scipy
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
tools
Add a jax_wheel Bazel rule to build jax pip packages
2024-10-24 14:20:46 -07:00
__init__.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
ad_checkpoint.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
api_util.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
BUILD
Renamed :pallas_gpu to :pallas_triton
2024-10-07 05:44:00 -07:00
cloud_tpu_init.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
collect_profile.py
Use pathlib for profiler log_dir
2024-07-28 18:47:45 +03:00
core.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
custom_batching.py
Use PEP484-style exports in several submodules
2024-08-14 08:59:56 -07:00
custom_derivatives.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
custom_transpose.py
Use PEP484-style exports in several submodules
2024-08-14 08:59:56 -07:00
debug.py
Explicitly re-export all names from jax.debug
2023-04-07 07:24:59 -07:00
distributed.py
Use PEP484-style exports in several submodules
2024-08-14 08:59:56 -07:00
dlpack.py
Use PEP484-style exports in several submodules
2024-08-14 08:59:56 -07:00
dtypes.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
errors.py
Add jax.errors.JaxRuntimeError as a public alias for the XlaRuntimeError class.
2024-09-26 08:39:30 -07:00
export.py
[export] Add support for serialization for some custom PyTree nodes
2024-10-21 11:38:13 +02:00
flatten_util.py
…
monitoring.py
Allow duration event listeners to take extra keyword arguments.
2023-10-04 13:15:49 -07:00
profiler.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
py.typed
…
random.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
sharding.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
stages.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
test_util.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
tree_util.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
tree.py
Add jax.tree module with aliases of jax.tree_util
2024-02-12 13:07:59 -08:00
typing.py
Fix typo in jax.typing module doc
2024-09-11 23:34:03 +10:00
util.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
version.py
Merge branch 'release/0.4.35'
2024-10-23 08:50:31 -04:00