jax
CI: 12/10/24 upstream sync
#182
Merged

CI: 12/10/24 upstream sync #182

github-actions merged 13 commits into rocm-main from ci-upstream-sync-55_1
github-actions
hawkinsp Remove code in jax2tf for compatibility with TF 2.10 or earlier.
acae2f05
Google-ML-Automation [AutoPGLE] Cleanup compiler code.
8813973d
Google-ML-Automation Fix mypy failure
6dbafed7
Google-ML-Automation [AutoPGLE] Share FDO profile even when compilation cache disabled.
e6dfe8f3
beckerhe Reverts bdadc53ebcd40a5091d66d2586deba82fe5e01ca
cb6881d9
jakevdp jax.core: deprecate a number of APIs
6541a620
Google-ML-Automation Merge pull request #25378 from hawkinsp:optbarrier
210bd308
dfm Deduplicate some GPU plugin definition code.
593143e1
WindQAQ [Pallas] Add non-square pl.dot test cases.
e418e883
Google-ML-Automation Merge pull request #25357 from jakevdp:core-deps
9c32fe8f
bixia1 [jax:custom_partitioning] Make SdyShardingRule a user facing class.
d4899f7b
Delete non-public API jax.lib.xla_bridge._backends
e6d6c4ef
Merge remote-tracking branch 'origin/rocm-main' into ci-upstream-sync…
20c5c714
github-actions[bot] github-actions[bot] enabled auto-merge 1 year ago
charleshofer charleshofer requested a review from charleshofer charleshofer 1 year ago
charleshofer
charleshofer approved these changes on 2024-12-10
github-actions github-actions merged 6dc4deea into rocm-main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone