Jake VanderPlas d3b3cd369f refactor: move sorting ops out of lax_numpy

2025-02-07 08:18:04 -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

Replace Hidden/Visible/Collective AxisTypes names with Auto/Explicit/Manual.

2025-01-24 23:21:13 -08:00

fft.py

refactor: move scalar type defs out of lax_numpy.py

2025-02-06 14:48:10 -08:00

index_tricks.py

Set __module__ attribute for objects in jax.numpy

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

lax_numpy.py

refactor: move sorting ops out of lax_numpy

2025-02-07 08:18:04 -08:00

linalg.py

refactor: move scalar type defs out of lax_numpy.py

2025-02-06 14:48:10 -08:00

polynomial.py

internal: move more NumPy APIs to ensure_arraylike

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

reductions.py

doc: add note about f16 casting in jnp.mean

2025-02-05 10:46:07 -08:00

scalar_types.py

refactor: move scalar type defs out of lax_numpy.py

2025-02-06 14:48:10 -08:00

setops.py

refactor: move sorting ops out of lax_numpy

2025-02-07 08:18:04 -08:00

sorting.py

refactor: move sorting ops out of lax_numpy

2025-02-07 08:18:04 -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

[sharding_in_types] Upstream changes from defaulting sharding_in_types config to True experiment. There aren't a lot of failures in TGP but we can atleast upstream these changes until we work on the failures.

2025-01-28 11:04:42 -08:00

vectorize.py

Set __module__ attribute for objects in jax.numpy

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