carlosgmartin 1f114b1cf7 Add numpy.put_along_axis.

2024-11-14 15:23:26 -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_methods.py

Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on.

2024-10-29 11:04:31 -07:00

fft.py

Add lax.FftType.

2024-10-10 08:07:35 -07:00

index_tricks.py

Replaced most usages of abc.ABC with util.StrictABC

2024-01-29 12:40:43 +00:00

lax_numpy.py

Add numpy.put_along_axis.

2024-11-14 15:23:26 -05:00

linalg.py

issue #24691

2024-11-04 23:52:54 +08:00

polynomial.py

CI: enable additional ruff formatting checks

2024-10-16 16:09:54 -07:00

reductions.py

Fix debug_nans false positive in jnp.quantile

2024-11-05 15:36:14 -08:00

setops.py

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

ufunc_api.py

Add some missing jax.numpy documentation

2024-10-25 13:14:44 -07:00

ufuncs.py

Internal: create decorators for defining ufuncs

2024-11-11 09:02:36 -08:00

util.py

Remove obsolete implements() decorator & fix tests

2024-10-28 15:22:09 -07:00

vectorize.py

Test that jax.numpy docstrings include examples

2024-09-21 07:39:17 -07:00