jax authors 18f2f19c1a Merge pull request #26525 from wenscarl:e2m1fn ...

PiperOrigin-RevId: 735457804

2025-03-10 11:46:18 -07: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

Reverts f3fade3b70443b6cf87f01f360e6a1cb85d4b1fb

2025-02-27 03:26:37 -08:00

einsum.py

Auto complete specs in a sharding if aval.ndim > len(sharding.spec) with None. So that for a 2D input, P('data') continues to work.

2025-03-06 16:10:14 -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

Improved errors when indexing with floats

2025-02-28 15:04:07 -08:00

lax_numpy.py

Fix wrong results in multidimensional pad.

2025-03-03 15:25:08 -05:00

linalg.py

Fix linalg.norm to return zero for proper norms of empty matrices.

2025-03-03 15:02:34 -05: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

support e2m1fn

2025-03-05 17:44:34 +00:00

setops.py

Update jnp.unique to support upstream interface changes.

2025-03-04 05:24:52 -05: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

jax.numpy ndim/shape/size: deprecate non-array input

2025-03-04 10:42:32 -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