transformers
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu")
#27593

Merged

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu") #27593

ArthurZucker merged 1 commit into huggingface:main from hvaara:jax-local-cpu

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backe…

311f993c

hvaara marked this pull request as ready for review 2 years ago

sanchit-gandhi approved these changes on 2023-12-01

sanchit-gandhi requested a review from

ArthurZucker 2 years ago

ArthurZucker approved these changes on 2023-12-04

ArthurZucker merged bcd0a91a into main 2 years ago

hvaara deleted the jax-local-cpu branch 2 years ago

Reviewers

ArthurZucker

sanchit-gandhi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone