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

Merge pull request #18614 from mattjj:custom-gradient-in-readthedocs

Browse Source 
PiperOrigin-RevId: 586074376
This commit is contained in:
jax authors 2023-11-28 12:47:03 -08:00
commit 8020e7d535
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/jax.rst
View File

@ -99,6 +99,7 @@ Automatic differentiation
vjp
custom_jvp
custom_vjp
custom_gradient
closure_convert
checkpoint