Jake VanderPlas a6a0226a53 jnp.linalg.norm: better documentation & error text for axis

2025-01-27 10:39:19 -08:00

..

__init__.py

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

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

array_api_metadata.py

[array api] Finalize array API in jax.numpy & deprecate jax.experimental.array_api

2024-08-01 11:19:17 -07:00

array_methods.py

Don't pass dtype to lax_internal._zero

2025-01-24 06:06:38 -08:00

fft.py

internal: move more NumPy APIs to ensure_arraylike

2025-01-23 08:48:13 -08:00

index_tricks.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00

lax_numpy.py

Add e8m0fnu support by conditional dtype.

2025-01-22 21:57:43 +00:00

linalg.py

jnp.linalg.norm: better documentation & error text for axis

2025-01-27 10:39:19 -08:00

polynomial.py

internal: move more NumPy APIs to ensure_arraylike

2025-01-23 08:48:13 -08:00

reductions.py

Avoid calls to warnings.catch_warnings in JAX core code.

2024-12-20 15:43:03 -05:00

setops.py

jax.numpy setops: use ensure_arraylike & avoid asarray

2025-01-21 16:05:49 -08:00

ufunc_api.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00

ufuncs.py

enable partitionable threefry by default

2025-01-13 22:46:24 -08:00

util.py

internal: check integer overflow in lax.asarray

2025-01-17 14:38:13 -08:00

vectorize.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00