mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 04:16:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax
History
Peter Hawkins f7eef2eda8 Use the upstream MLIR strip-debuginfo pass instead of hand-rolling our own. 
(I had missed that the upstream pass exists!)

Fixes https://github.com/google/jax/issues/16649

PiperOrigin-RevId: 548192839
2023-07-14 12:24:59 -07:00
..
_src
Use the upstream MLIR strip-debuginfo pass instead of hand-rolling our own.
2023-07-14 12:24:59 -07:00
example_libraries
Use lower-case PEP 585 names for types.
2023-06-23 15:12:14 -07:00
experimental
Merge pull request #16718 from mattjj:scatter-apply-autodiff
2023-07-14 09:24:12 -07:00
extend
initiate jax.extend via docs and top-level module set-up
2023-05-15 15:47:06 -07:00
image
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
interpreters
[shape_poly] More cleanup for the internal APIs for shape polymorphism.
2023-07-13 16:37:53 +03:00
lax
deprecate jax.lax.prod
2023-05-23 17:33:50 -07:00
lib
[JAX] Move jax._src.lib.xla_bridge to jax._src.xla_bridge.
2023-02-28 07:01:57 -08:00
nn
Add test framework for module attribute
2023-04-21 13:20:16 -07:00
numpy
Remove several deprecated APIs
2023-07-11 12:42:32 -07:00
ops
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
scipy
Add jax.scipy.stats.binom
2023-06-27 03:41:38 -07:00
tools
Import py_binary() rule from @rules_python.
2023-07-14 08:09:07 -07:00
__init__.py
Remove several deprecated APIs
2023-07-11 12:42:32 -07:00
abstract_arrays.py
Mark jax.abstract_arrays as deprecated
2023-06-07 23:36:40 -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
Remove the old JAX/IREE integration.
2023-07-11 10:10:52 -04: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
[shape_poly] Cleanup exported API symbols.
2023-07-03 23:10:07 -07: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
custom_vjp symbolic zeros support, take two
2023-04-05 11:17:05 -07: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
Explicitly re-export all names from jax.debug
2023-04-07 07:24:59 -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
Remove jax.dtypes._jax_types
2023-02-01 12:49:06 -08:00
errors.py
errors: create TracerBoolConversionError for more targeted debugging tips
2023-06-21 01:41:45 -07:00
flatten_util.py
ravel_pytree now accepted by static linters
2023-03-07 23:35:48 +00:00
jaxpr_util.py
Use lower-case PEP 585 names for types.
2023-06-23 15:12:14 -07:00
linear_util.py
jax.linear_util: remove unnecessary exported names
2022-12-21 18:52:01 -08:00
monitoring.py
Move contents of jax.monitoring to jax._src.monitoring.
2023-02-08 15:03:22 -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
add new random key constructor
2023-05-22 11:35:10 -07:00
sharding.py
Delete OpShardingSharding export since it has been 3 months since it was deprecated. Also remove deprecation warnings for MeshPspecSharding.
2023-06-08 13:53:38 -07:00
stages.py
add CompilerOptions type alias for AOT compiler options
2023-04-12 16:49:13 -07:00
test_util.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
tree_util.py
Add jax.tree_util.tree_leaves_with_path(tree).
2023-06-12 04:13:37 -07:00
typing.py
jax.typing: recommend instance check in Python 3.10 or newer
2023-03-27 10:01:28 -07:00
util.py
Document why 'import name as name' is used
2022-12-14 15:07:04 -08:00
version.py
Update version numbers and changelog post 0.4.13 release
2023-06-22 17:54:57 -07:00