jax
e459d3bb
- Deprecate jax.lax.infeed and jax.lax.outfeed.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Deprecate jax.lax.infeed and jax.lax.outfeed. These APIs are already broken on GPU and TPU by virtue of not being implemented in the PJRT C API, so it seems unlikely that they have any users. PiperOrigin-RevId: 746595857
References
#392 - CI: 04/30/25 upstream sync
#518 - [0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test
#519 - [0.6.0-UT] Fix Test Shard
#594 - Add CI workflow for Pytest ROCm
Author
hawkinsp
Committer
charleshofer
Parents
100eb7c0
Loading