jax
c55f1876 - Remove code that guards against int4 not being available.

Commit
1 year ago
Remove code that guards against int4 not being available. JAX depends on ml_dtypes >= 0.2, and int4 was added in ml_dtypes 0.2. PiperOrigin-RevId: 607345852
Author
Committer
Parents
Loading