Jake VanderPlas 33b989ac9e refactor: import numpy objects directly in jax.numpy

2025-02-14 12:47:58 -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_creation.py

refactor: move array creation routines out of lax_numpy.py

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

array_methods.py

Merge pull request #26498 from jakevdp:jnp-indexing

2025-02-14 07:16:00 -08:00

einsum.py

Merge pull request #26472 from jakevdp:jnp-einsum

2025-02-13 11:55:07 -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

Fix the type annotations and don't += a generator (it's confusing)

2025-02-14 12:46:01 -08:00

lax_numpy.py

refactor: import numpy objects directly in jax.numpy

2025-02-14 12:47:58 -08: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

Removed unused `# type: ignore` comments

2025-02-13 21:12:27 +00: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