mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 04:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
Skye Wanderman-Milne 66ba734882
Add note to docs describing how pytree arguments work. (#3284) 
Addresses #3095. I'm not sure if we wanna link to this from API
docstrings.

This also subsumes the original pytrees notebook.
2020-06-03 09:46:00 -07:00
..
autodiff_cookbook.ipynb
autodiff cookbook: assume continuous second derivatives
2020-04-21 14:08:26 -07:00
Common_Gotchas_in_JAX.ipynb
mention numpy & scipy convolve functions in gotchas doc. (#3214)
2020-05-27 13:20:27 -07:00
Custom_derivative_rules_for_Python_code.ipynb
Fix sign error in examples. (#3031)
2020-05-11 08:52:55 -07:00
How_JAX_primitives_work.ipynb
Use a whitelist to restrict visibility in top-level jax namespace. (#2982)
2020-05-07 17:24:19 -04:00
maml.ipynb
Moved all notebooks to docs/notebooks. (#1493)
2019-10-17 08:58:25 +02:00
Neural_Network_and_Data_Loading.ipynb
Correct calculation of loss and increase learning rate (#3113)
2020-05-19 15:18:34 -07:00
neural_network_with_tfds_data.ipynb
Correct calculation of loss and increase learning rate (#3113)
2020-05-19 15:18:34 -07:00
quickstart.ipynb
Added link to README (#3139)
2020-05-18 14:12:52 -07:00
README.md
Added more documentation for how to fix notebook build failures (#2404)
2020-03-12 10:59:30 +01:00
score_matching.ipynb
Fix Typos
2019-12-30 11:27:12 -08:00
vmapped_log_probs.ipynb
Remove unnecessary code in colabs (#2623)
2020-04-06 17:26:51 -07:00
Writing_custom_interpreters_in_Jax.ipynb
Added type annotations and comments related to partial evaluation.
2020-04-09 13:00:33 +03:00
XLA_in_Python.ipynb
Moved all notebooks to docs/notebooks. (#1493)
2019-10-17 08:58:25 +02:00

README.md

For instructions on how to change and test notebooks, see Update Documentation.