Yash Katariya 12b59f8e53 Rename hidden_mode -> hidden_axes and hidden_mode_ctx -> use_hidden_axes. Same for visible mode and visible_mode_ctx. ...

Also make the `axes` parameter optional of hidden_axes and visible_axes functions. If axes is optional, you drop into full hidden/visible mode.

PiperOrigin-RevId: 716771872

2025-01-17 13:01:07 -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

Rename hidden_mode -> hidden_axes and hidden_mode_ctx -> use_hidden_axes. Same for visible mode and visible_mode_ctx.

2025-01-17 13:01:07 -08:00

fft.py

Add support for N-D FFTs with D>3.

2024-12-19 15:23:30 +00:00

index_tricks.py

Set __module__ attribute for objects in jax.numpy

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

lax_numpy.py

Merge pull request #25936 from jakevdp:ensure-arraylike

2025-01-17 10:23:14 -08:00

linalg.py

Use ensure_arraylike utility in jax.numpy.linalg

2025-01-17 11:00:31 -08:00

polynomial.py

Fix type annotations for NumPy 2.2

2024-12-11 14:24:58 -08:00

reductions.py

Avoid calls to warnings.catch_warnings in JAX core code.

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

setops.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -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

Add ensure_arraylike utility for lax.numpy implementations

2025-01-16 16:46:11 -08:00

vectorize.py

Set __module__ attribute for objects in jax.numpy

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