mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 23:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax
History
Alex Minnaar 0893b08a6f histogramdd implemenation
2020-10-01 18:02:33 -04:00
..
experimental
Merge pull request #4343 from SIben:test_cholesky
2020-09-19 09:25:44 -07:00
image
Fix jax.image._resize function (#3805)
2020-07-20 13:15:40 -07:00
interpreters
fix typo
2020-09-18 19:41:59 -07:00
lax
Merge pull request #4330 from google:create-token-stop-grad
2020-09-18 18:27:04 -07:00
lib
Fix code quality issues (#4302)
2020-09-17 09:21:18 -07:00
nn
omnistaging on by default (#4038)
2020-09-15 08:06:46 -07:00
numpy
histogramdd implemenation
2020-10-01 18:02:33 -04:00
ops
Fixing weird behavior in segment_sum when num_segments is None (#4034)
2020-09-11 13:51:42 -04:00
scipy
Make expm transposable and remove custom_jvp rule. (#4314)
2020-09-17 21:10:54 -07:00
third_party
Cleanup: avoid jnp.prod & np.prod on array shapes (#4086)
2020-08-18 10:17:38 -07:00
tools
Fix code quality issues (#4302)
2020-09-17 09:21:18 -07:00
__init__.py
Add jax.linear_transpose (#3398)
2020-09-16 20:29:19 -07:00
abstract_arrays.py
Fix code quality issues (#4302)
2020-09-17 09:21:18 -07:00
ad_util.py
omnistaging, under a flag and disabled by default (#3370)
2020-07-30 12:59:36 -07:00
api_util.py
fix typo
2020-09-18 19:41:59 -07:00
api.py
plumb donate_argnums into jax.xla_computation
2020-09-18 17:39:05 -07:00
BUILD
Add jit to jax.image.resize (#3714)
2020-07-10 10:32:13 -04:00
config.py
omnistaging on by default (#4038)
2020-09-15 08:06:46 -07:00
core.py
rename and simplify TypedJaxpr -> ClosedJaxpr (#4328)
2020-09-18 10:07:13 -07:00
custom_derivatives.py
rename and simplify TypedJaxpr -> ClosedJaxpr (#4328)
2020-09-18 10:07:13 -07:00
dlpack.py
Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)
2020-05-11 17:43:55 -04:00
dtypes.py
Avoid direct type/dtype comparisons to fix NumPy 1.19 deprecation war… (#3543)
2020-06-24 15:19:00 -04:00
flatten_util.py
deflake
2020-06-12 15:41:49 -07:00
jaxpr_util.py
move experimental.jaxpr_stats to jaxpr_util
2020-08-18 18:07:38 -07:00
lax_linalg.py
Fix conditional in eig and expand eig test suite. (#4320)
2020-09-18 10:30:19 +03:00
lax_reference.py
refine population_count type check (#3887)
2020-07-28 19:46:00 -07:00
lazy.py
deflake remainder of jax (#3343)
2020-06-06 10:51:34 -07:00
linear_util.py
Interrupt lu transformation generators whenever an exception occurs
2020-09-09 20:43:05 +02:00
pprint_util.py
Attach source info to Jaxpr equations. (#3421)
2020-06-17 16:35:36 -07:00
profiler.py
Add a jax.profiler.StepTraceContext API. (#3591)
2020-07-10 17:20:06 -07:00
random.py
[jax2tf] Fix random.split when jax_exable_x64 (#4208)
2020-09-07 14:41:50 +03:00
source_info_util.py
Attach source info to Jaxpr equations. (#3421)
2020-06-17 16:35:36 -07:00
test_util.py
Fix code quality issues (#4302)
2020-09-17 09:21:18 -07:00
traceback_util.py
trivial change to test source sync
2020-09-18 16:09:41 -07:00
tree_util.py
Compare treedefs by num_leaves not traversal_ in tree_transpose. (#3659)
2020-07-30 13:31:17 -04:00
util.py
Add support for negative axes to vmap. (#4111)
2020-08-24 20:21:19 -04:00
version.py
update version and changelog for pypi (#4294)
2020-09-15 08:00:47 -07:00