Sai-Suraj-27 29def4eefa Updated all the pre-commit hooks versions.

2024-04-08 00:59:02 +05:30

..

263-prng.md

DOC: read-through and edit the new jax tutorials

2024-03-20 18:18:08 -07:00

2026-custom-derivatives.md

Create JAX Enhancement Proposals (JEPs).

2022-08-08 16:13:58 -04:00

4008-custom-vjp-update.md

convert custom VJP update guide to a retroactive JEP

2022-08-10 13:45:57 -07:00

4410-omnistaging.md

Create JAX Enhancement Proposals (JEPs).

2022-08-08 16:13:58 -04:00

9263-typed-keys.md

Change vmap argument from axis to in_axes for Inadvertent transposing of key buffers in 9263-typed-keys.md

2024-03-28 15:10:52 +05:30

9407-type-promotion.ipynb

Fixed Typos in JEP doc files

2024-03-12 09:58:13 +05:30

9407-type-promotion.md

Updated all the pre-commit hooks versions.

2024-04-08 00:59:02 +05:30

9419-jax-versioning.md

Deprecate support for the mhlo dialect.

2024-01-15 02:13:40 -08:00

10657-sequencing-effects.md

Fixed Typos in JEP doc files

2024-03-12 09:58:13 +05:30

11830-new-remat-checkpoint.md

delete old remat implementation

2022-08-16 23:16:37 -07:00

12049-type-annotations.md

fix: typo inside docs/jep/12049-type-annotations.md

2023-10-13 08:51:01 +05:45

14273-shard-map.md

Deprecate support for the mhlo dialect.

2024-01-15 02:13:40 -08:00

15856-jex.md

JEX: move jax.linear_util to jax.extend.linear_util

2023-08-30 18:32:12 -07:00

17111-shmap-transpose.md

Fix typos across the package

2023-09-22 14:54:31 -07:00

18137-numpy-scipy-scope.md

Fixed Several Typos

2024-03-12 00:37:46 +05:30

index.rst

JEP 18137: Scope of JAX NumPy & SciPy Wrappers

2023-11-03 19:19:21 -07:00