Sergei Lebedev 194884d311 Migrated to mypy 1.14.1 with --allow_redefinition ...

I initially wanted to upgrade to 1.15, but it seems to have a bug in how
ternary expressions are type checked. For example,

   def f(x: int) -> str: ...
   def g(x: int) -> str: ...

   callback = f if ... else g  # has type object!

2025-02-13 15:38:28 +00: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

refactor: move lax_numpy tensor contractions into their own file

2025-02-10 18:56:18 -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

Merge pull request #26447 from jakevdp:refactor-contractions

2025-02-12 07:14:30 -08:00

linalg.py

refactor: move lax_numpy tensor contractions into their own file

2025-02-10 18:56:18 -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 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

tensor_contractions.py

refactor: move lax_numpy tensor contractions into their own file

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

ufunc_api.py

Migrated to mypy 1.14.1 with --allow_redefinition

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

ufuncs.py

enable partitionable threefry by default

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

util.py

Migrated to mypy 1.14.1 with --allow_redefinition

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

vectorize.py

Set __module__ attribute for objects in jax.numpy

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

window_functions.py

refactor: move lax_numpy window functions into their own file

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