jax
CI: 01/07/25 upstream sync
#194
Merged

CI: 01/07/25 upstream sync #194

github-actions merged 32 commits into rocm-main from ci-upstream-sync-80_1
github-actions
dfm Move jex.ffi to jax.ffi.
cb4d97aa
Ruturaj4 Update package indentation fix
20b75ab8
Google-ML-Automation Merge pull request #25715 from ROCm:ci_build_code_fixes-upstream
4de7794f
dfm Temporarily allow deprecation warnings for `scipy.special.lpmn` and `…
512d5450
belitskiy Disable `avxvnniint8` when building with Clang version < 19, or GCC <…
f2e210b3
gnecula [shape_poly] Remove old non_negative support.
e87a2a59
yliu120 fix the degenerated case
3ff000ee
marksandler2 Fixes tril/triu comments (they were flipped)
6c87bf38
jakevdp Deprecate scipy.special.lpmn & lpmn_values
245a13a3
Google-ML-Automation Update XLA dependency to use revision
18b193cb
Google-ML-Automation Merge pull request #25675 from jakevdp:dep-lpmn
74be8bd9
maxwillzq bazel: export serialization.fbs for downstream usage
c39e38fe
jakevdp jnp.einsum: default to optimize='auto'
2f7204ff
Google-ML-Automation Merge pull request #25699 from yliu120:fix_iota
634b45bf
bythew3i [Mosaic TPU] Validate inserted layout in relayout-insertion pass.
9f842909
Google-ML-Automation Merge pull request #25214 from jakevdp:einsum-optimize
52cc5c7f
hawkinsp Suppress MSAN warnings from SVD that are showing up in CI.
61dd0412
hawkinsp Rename pybind_extension to nanobind_extension.
90d8f378
zhenying-liu fix the doc error: module 'scipy.misc' has no attribute 'face'
77c6947a
pschuh Remove the need for check_rep for with_sharding_constraint.
b49ba655
sharadmv [Mosaic TPU] Add some elementwise canonicalizations
4caa263a
jakevdp Remove deprecated jax.experimental.array_api
c7b0d681
Google-ML-Automation Merge pull request #25740 from jakevdp:remove-array-api
b304b9ef
yashk2810 Make inspect_array_sharding work without mesh context manager too.
23eaf216
Google-ML-Automation Merge pull request #25728 from zhenying-liu:scipy.misc
7997f080
gnecula [shape_poly] Improve threefry with symbolic shapes
bc3306c8
shoyer Fix type signature for __divmod__
7fb68cac
Google-ML-Automation Merge pull request #25731 from gnecula:poly_random_even
712bece2
Google-ML-Automation Merge pull request #25633 from dfm:move-ffi
56f0f953
Google-ML-Automation Merge pull request #25748 from shoyer:divmod
853af560
dfm Add a register_custom_type_id function to the GPU plugins.
a7f384cc
github-actions[bot] github-actions[bot] enabled auto-merge 1 year ago
charleshofer Unskip unit tests that are now fixed
a94ee1fc
charleshofer
charleshofer approved these changes on 2025-01-07
github-actions github-actions merged 972f95b9 into rocm-main 1 year ago
charleshofer charleshofer deleted the ci-upstream-sync-80_1 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone