transformers/templates
Roy Hvaara fc63914399
[JAX] Replace uses of jnp.array in types with jnp.ndarray. (#26703)
`jnp.array` is a function, not a type:
https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.array.html
so it never makes sense to use `jnp.array` in a type annotation. Presumably the intent was to write `jnp.ndarray` aka `jax.Array`.

Co-authored-by: Peter Hawkins <phawkins@google.com>
2023-10-10 21:35:16 +02:00
..
adding_a_missing_tokenization_test Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
adding_a_new_example_script Fix token in example template (#25351) 2023-08-08 12:00:31 +02:00
adding_a_new_model [JAX] Replace uses of jnp.array in types with jnp.ndarray. (#26703) 2023-10-10 21:35:16 +02:00