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

Merged

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

amyeroberts merged 9 commits into huggingface:main from gojiteji:jaxnumpy-jit-compatibility

Replace numpy operations with jax.numpy for JIT compatibility

96e9dd42

rm numpy import

848f8df1

sanchit-gandhi approved these changes on 2023-05-15

sanchit-gandhi requested a review from

amyeroberts 2 years ago

rm numpy import and fix np->jnp

175ab5dd

sanchit-gandhi commented on 2023-05-15

fixed slices bug

d110c34d

amyeroberts approved these changes on 2023-05-15

fixed decoder_start_tokens -> decoder_start_token_id

f25d5881

fixed jnp in modleing mt5

c74abb81

doc fix

33b5904e

rm numpy import

b6b5e67b

make

c90229e0

gojiteji requested a review from

amyeroberts 2 years ago

amyeroberts approved these changes on 2023-05-16

amyeroberts merged ba6815e8 into main 2 years ago

Reviewers

amyeroberts

sanchit-gandhi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone