transformers
Replace NumPy Operations with JAX NumPy Equivalents for JIT Compilation Compatibility
#23356
Merged
Merged