Peter Hawkins 66293d8897 Remove code present to support jaxlib < 0.5.1. ...

The new minimum xla_extension_version is 317 and the new mlir_api_version is 58.

2025-02-26 07:40:40 -05: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_creation.py

refactor: move array creation routines out of lax_numpy.py

2025-02-06 15:47:30 -08:00

array_methods.py

[sharding_in_types] Make slice and ellipsis work with .at[...].get(out_sharding=P(...))

2025-02-21 18:25:11 -08:00

einsum.py

Error out if going from Manual -> Auto/Explicit AxisTypes in the auto_axes and explicit_axes API that do mesh_cast implicitly.

2025-02-19 09:21:53 -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

indexing.py

[sharding_in_types] Make slice and ellipsis work with .at[...].get(out_sharding=P(...))

2025-02-21 18:25:11 -08:00

lax_numpy.py

Remove code present to support jaxlib < 0.5.1.

2025-02-26 07:40:40 -05:00

linalg.py

refactor: import numpy objects directly in jax.numpy

2025-02-14 12:47:58 -08:00

polynomial.py

Migrated to mypy 1.14.1 with --allow_redefinition

2025-02-13 15:38:28 +00:00

reductions.py

jax.numpy reductions: avoid upcast of f16 when dtype is specified by user

2025-02-12 11:49:39 -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 lax_numpy indexing routines to their own submodule

2025-02-13 12:03:07 -08:00

sorting.py

refactor: move sorting ops out of lax_numpy

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

tensor_contractions.py

refactor: move lax_numpy tensor contractions into their own file

2025-02-10 18:56:18 -08:00

ufunc_api.py

refactor: move lax_numpy indexing routines to their own submodule

2025-02-13 12:03:07 -08:00

ufuncs.py

enable partitionable threefry by default

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

util.py

Now that sharding_in_types config flag is True, remove the config and all the conditionals

2025-02-19 06:53:35 -08:00

vectorize.py

refactor: import numpy objects directly in jax.numpy

2025-02-14 12:47:58 -08:00

window_functions.py

refactor: move lax_numpy window functions into their own file

2025-02-07 11:21:38 -08:00