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
History
Jake VanderPlas 822b6aad3b jax.scipy.qr: fix return type for mode='r'
2022-04-26 11:26:56 -07:00
..
_src
jax.scipy.qr: fix return type for mode='r'
2022-04-26 11:26:56 -07:00
example_libraries
Rename jax.nn.normalize to standardize. Add normalize alias with DeprecationWarning.
2022-03-23 20:55:22 +00:00
experimental
Add CustomCall MLIR lowering for HCB outside_call primitive
2022-04-25 19:38:39 -07:00
image
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
interpreters
[remove-units] prevent cond partial eval from introducing units
2022-04-24 14:28:33 -07:00
lax
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
lib
Use imports relative to the jax package consistently, rather than .-relative imports.
2021-11-24 07:48:29 -08:00
nn
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
numpy
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
ops
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
scipy
implement scipy.cluster.vq.vq
2022-04-23 03:14:32 +08:00
tools
Fix SyntaxError in jax_to_ir
2021-12-21 15:11:08 -08:00
__init__.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
abstract_arrays.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
ad_checkpoint.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
api_util.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
cloud_tpu_init.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
config.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
core.py
Merge pull request #10371 from sharadmv:mlir-token-lowering
2022-04-21 13:08:21 -07:00
custom_batching.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
custom_derivatives.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
custom_transpose.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
distributed.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
dlpack.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
dtypes.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
errors.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
flatten_util.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
jaxpr_util.py
Increase minimum jaxlib version to 0.3.0.
2022-03-04 10:33:03 -05:00
linear_util.py
prototyping dynamic shapes
2022-04-11 22:10:47 -07:00
prng.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
profiler.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
py.typed
Improve type annotations
2021-01-13 10:26:35 -05:00
random.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
stages.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
test_util.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
tree_util.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
util.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
version.py
Increase minimum jaxlib version to 0.3.7.
2022-04-18 08:09:50 -04:00