PR #35919 Update deprecated Jax calls

Update deprecated Jax calls #35919

ArthurZucker merged 4 commits into huggingface:main from rasmi:patch-3

Rocketknight1 added Flax

rasmi force pushed from 273d76e4 to e3d7298c 1 year ago

rasmi changed the title ~~Update deprecated arguments in jax.numpy.clip.~~ Update deprecated Jax calls 1 year ago

ArthurZucker commented on 2025-02-10

Remove deprecated arguments for jax.numpy.clip.

004c2e37

Remove deprecated arguments for jax.numpy.clip.

ee0aab52

Update jax version to 0.4.27 to 0.4.38.

4f0c56c2

Avoid use of deprecated xla_bridge.get_backend().platform

d0b139d4

rasmi force pushed from 52d10f6c to d0b139d4 1 year ago

rasmi requested a review from

ArthurZucker 1 year ago

ArthurZucker approved these changes on 2025-03-20

ArthurZucker merged f0d5b2ff into main 1 year ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

Flax

Milestone

No milestone