xla
[Pallas] Support programmatically extracting the payload
#6696

Merged

[Pallas] Support programmatically extracting the payload #6696

alanwaketan merged 11 commits into master from alanwaketan/pallas_jax

alanwaketan requested a review from

will-cromar 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

alanwaketan requested a review from

qihqi 1 year ago

will-cromar commented on 2024-03-08

alanwaketan force pushed from 7ec49031 to 769fa669 1 year ago

JackCaoG approved these changes on 2024-03-08

will-cromar approved these changes on 2024-03-08

[Pallas] Support programmatically extracting the payload

c7a60b18

Fix linters

acfc791d

install jax in tpu ci

cd5fe098

Skip the test for non tpu ci

2efae7c9

Move tests

a7e7c2a0

Fix linters

f63e3c9f

Improve jax dependency

3f05b90f

Fix comment

18841a43

alanwaketan force pushed from 0540fb8c to 18841a43 1 year ago

Fix a type

07ad7390

Add tpu ci

95075ed8

Try fixing tpu ci

ae6f8dc9

alanwaketan merged 37067917 into master 1 year ago

alanwaketan deleted the alanwaketan/pallas_jax branch 1 year ago

vanbasten23 restored the head branch 1 year ago

Reviewers

will-cromar

JackCaoG

qihqi

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone

xla [Pallas] Support programmatically extracting the payload #6696 Merged

[Pallas] Support programmatically extracting the payload #6696

xla
[Pallas] Support programmatically extracting the payload
#6696

Merged