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
harryjulian c0d4ae0cc3 Added scipy.stats.bernoulli cdf and ppf.
2022-12-22 18:12:25 +00:00
..
_src
Added scipy.stats.bernoulli cdf and ppf.
2022-12-22 18:12:25 +00:00
example_libraries
Revert part of #13498
2022-12-05 09:21:09 -08:00
experimental
(NFC) Prepare for migration from producing MHLO to producing StableHLO
2022-12-15 21:00:07 -08:00
image
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
interpreters
(NFC) Prepare for migration from producing MHLO to producing StableHLO
2022-12-15 21:00:07 -08:00
lax
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
lib
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
nn
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
numpy
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
ops
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
scipy
Added scipy.stats.bernoulli cdf and ppf.
2022-12-22 18:12:25 +00:00
tools
Update values for release 0.4.1
2022-12-12 19:04:38 -08:00
__init__.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
abstract_arrays.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
ad_checkpoint.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
api_util.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
BUILD
Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.
2022-11-28 14:31:48 -08:00
cloud_tpu_init.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
collect_profile.py
Cleanup: remove unused imports in private modules
2022-10-20 14:37:21 -07:00
config.py
Cleanup: remove unused imports in private modules
2022-10-20 14:37:21 -07:00
core.py
[jax] move jax.core to jax._src.core
2022-12-15 20:35:20 -08:00
custom_batching.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
custom_derivatives.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
custom_transpose.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
debug.py
Add inspect_array_sharding, enabling looking at shardings in pjit-ted functions
2022-09-22 17:36:56 -07:00
distributed.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
dlpack.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
dtypes.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
errors.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
flatten_util.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
jaxpr_util.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
linear_util.py
add a basic prototype of piles, behind jax_dynamic_shapes
2022-11-06 17:03:04 -08:00
monitoring.py
Add JAX monitoring library that instruments code via events.
2022-11-15 12:41:41 -08:00
prng.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
profiler.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
py.typed
Improve type annotations
2021-01-13 10:26:35 -05:00
random.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
sharding.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
stages.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
test_util.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
tree_util.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
util.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
version.py
Finish jax and jaxlib 0.4.2 release
2022-12-13 10:51:13 -08:00