Jake VanderPlas b18ca05bc6 jnp.mean: for f16 inputs, accumulate in f32

2023-09-27 18:27:19 -07:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

array_methods.py

Fix typos across the package

2023-09-22 14:54:31 -07:00

fft.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

index_tricks.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

lax_numpy.py

Deprecate non-array inputs to several jax.numpy functions

2023-09-22 14:21:23 -07:00

linalg.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

polynomial.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

reductions.py

jnp.mean: for f16 inputs, accumulate in f32

2023-09-27 18:27:19 -07:00

setops.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

ufunc_api.py

[typing] better annotations for jnp.ufunc

2023-09-21 12:48:14 -07:00

ufuncs.py

Merge pull request #16419 from mattjj:pow-jvp

2023-08-22 17:15:04 -07:00

util.py

Deprecate non-array inputs to several jax.numpy functions

2023-09-22 14:21:23 -07:00

vectorize.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00