mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #11838 from froystig:custom-vjp-upgrade-jep-move

Browse Source 
PiperOrigin-RevId: 466838413
This commit is contained in:
jax authors 2022-08-10 18:57:01 -07:00
commit 104e87f358
3 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/index.rst
View File

@ -64,7 +64,6 @@ parallelize, Just-In-Time compile to GPU/TPU, and more.
profiling
device_memory_profiling
rank_promotion_warning
custom_vjp_update
.. toctree::
:maxdepth: 2

0
docs/custom_vjp_update.md → docs/jep/4008-custom-vjp-update.md
View File

4
docs/jep/index.rst
View File

@ -39,6 +39,7 @@ Then create a pull request that adds a file named
263: JAX PRNG Design <263-prng>
2026: Custom JVP/VJP rules for JAX-transformable functions <2026-custom-derivatives>
4008: Custom VJP and `nondiff_argnums` update <4008-custom-vjp-update>
4410: Omnistaging <4410-omnistaging>
9407: Design of Type Promotion Semantics for JAX <9407-type-promotion>
9419: Jax and Jaxlib versioning <9419-jax-versioning>
@ -46,3 +47,6 @@ Then create a pull request that adds a file named
11830: `jax.remat` / `jax.checkpoint` new implementation <11830-new-remat-checkpoint>
Several early JEPs were converted in hindsight from other documentation,
issues, and pull requests, so they might not exactly reflect the process
outlined above.