rocm_jax

mirrors/rocm_jax

Fork 0

mirror of https://github.com/ROCm/jax.git synced 2025-04-24 19:56:05 +00:00

History

Jake VanderPlas 7545aeb8d3 doc build: avoid executing thinking_in_jax.ipynb

2021-02-04 09:26:11 -08:00

autodiff_cookbook.ipynb

Update autodiff_cookbook.ipynb

2020-12-07 10:06:08 -08:00

Common_Gotchas_in_JAX.ipynb

Update Common_Gotchas_in_JAX.ipynb

2020-10-05 10:19:18 -07:00

Custom_derivative_rules_for_Python_code.ipynb

Internal change

2020-12-07 08:31:40 -08:00

How_JAX_primitives_work.ipynb

Change onp/np to np/jnp in docs & notebooks (#3760 )

2020-07-15 13:17:38 -07:00

maml.ipynb

Change onp/np to np/jnp in docs & notebooks (#3760 )

2020-07-15 13:17:38 -07:00

Neural_Network_and_Data_Loading.ipynb

Change onp/np to np/jnp in docs & notebooks (#3760 )

2020-07-15 13:17:38 -07:00

neural_network_with_tfds_data.ipynb

Change onp/np to np/jnp in docs & notebooks (#3760 )

2020-07-15 13:17:38 -07:00

quickstart.ipynb

Move example

2020-12-07 10:05:00 -08: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

Change onp/np to np/jnp in docs & notebooks (#3760 )

2020-07-15 13:17:38 -07:00

thinking_in_jax.ipynb

doc build: avoid executing thinking_in_jax.ipynb

2021-02-04 09:26:11 -08:00

vmapped_log_probs.ipynb

only pass hashable values as static args

2020-10-16 13:11:56 -07:00

Writing_custom_interpreters_in_Jax.ipynb

[JAX] Move jax.util to jax._src_util.

2021-01-11 14:21:07 -08:00

XLA_in_Python.ipynb

Updates XLA_in_Python notebook with the latest API.

2021-01-07 10:14:50 -08:00

README.md

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