mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax
History
Jake VanderPlas 8378d08fcd Remove deprecated array.tile() method
2022-08-23 07:36:59 -07:00
..
_src
Remove deprecated array.tile() method
2022-08-23 07:36:59 -07:00
example_libraries
Merge pull request #11542 from mattjj:remove-resnet50-example
2022-07-19 16:56:56 -07:00
experimental
roll-forward #11952, take 2
2022-08-22 13:57:31 -07:00
image
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
interpreters
roll-forward #11952, take 2
2022-08-22 13:57:31 -07:00
lax
Add copy_p to jax.lax namespace
2022-08-19 13:09:15 -07:00
lib
Reexport jaxlib.__version as jax.lib.__version__.
2022-04-28 10:25:06 -07:00
nn
Annotate nn.initializers
2022-08-04 23:17:40 -04:00
numpy
DOC: ensure that _wraps() generates correct links to wrapped functions
2022-07-21 11:12:35 -07:00
ops
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
scipy
Implementation of jax.scipy.stats.gaussian_kde
2022-06-28 15:17:12 -04:00
tools
Internal change
2022-08-18 11:39:07 -07:00
__init__.py
Expose pure callback and enable rank polymorphic callbacks
2022-08-17 10:56:42 -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
BUILD
Remove deprecated jax.experimental.optimizers
2022-08-09 08:50:59 -07:00
cloud_tpu_init.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
collect_profile.py
Internal change
2022-08-18 11:39:07 -07:00
config.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
core.py
roll-forward #11952, take 2
2022-08-22 13:57:31 -07:00
custom_batching.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
custom_derivatives.py
Copybara import of the project:
2022-08-17 22:40:58 -07:00
custom_transpose.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
debug.py
Add debugging tools under jax.debug and documentation
2022-07-28 20:07:26 -07:00
distributed.py
Unbreak jax.distributed initialization.
2022-06-07 11:10:17 -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
Fix or ignore some pytype errors.
2022-05-31 21:25:55 -07:00
linear_util.py
[dynamic-shapes] fix minor bint bugs
2022-07-19 16:38:40 -07:00
prng.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
profiler.py
jax.profiler: remove deprecated functions
2022-07-19 08:13:44 -07:00
py.typed
…
random.py
roll-forward #11952, take 2
2022-08-22 13:57:31 -07:00
stages.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
test_util.py
Internal change
2022-08-18 11:39:07 -07:00
tree_util.py
Remove deprecated function jax.tree_util.tree_multimap
2022-07-26 09:37:27 -07:00
util.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
version.py
Update version + changelog
2022-08-11 17:11:52 -07:00