https://github.com/google/jax
History
Tip revision: 805ea11001f0f4ad289d180590ac19f28213af96 authored by Peter Hawkins on 20 December 2022, 14:13:12 UTC
[XLA:Python] Drop support for classic HLO in the Python bindings.
Tip revision: 805ea11
File Mode Size
Common_Gotchas_in_JAX.ipynb -rw-r--r-- 60.4 KB
Common_Gotchas_in_JAX.md -rw-r--r-- 36.1 KB
Custom_derivative_rules_for_Python_code.ipynb -rw-r--r-- 112.9 KB
Custom_derivative_rules_for_Python_code.md -rw-r--r-- 35.2 KB
Distributed_arrays_and_automatic_parallelization.ipynb -rw-r--r-- 71.1 KB
Distributed_arrays_and_automatic_parallelization.md -rw-r--r-- 27.5 KB
How_JAX_primitives_work.ipynb -rw-r--r-- 59.4 KB
How_JAX_primitives_work.md -rw-r--r-- 25.7 KB
Neural_Network_and_Data_Loading.ipynb -rw-r--r-- 18.8 KB
Neural_Network_and_Data_Loading.md -rw-r--r-- 9.0 KB
README.md -rw-r--r-- 158 bytes
Writing_custom_interpreters_in_Jax.ipynb -rw-r--r-- 16.4 KB
Writing_custom_interpreters_in_Jax.md -rw-r--r-- 11.4 KB
autodiff_cookbook.ipynb -rw-r--r-- 56.4 KB
autodiff_cookbook.md -rw-r--r-- 37.5 KB
convolutions.ipynb -rw-r--r-- 514.1 KB
convolutions.md -rw-r--r-- 15.1 KB
external_callbacks.ipynb -rw-r--r-- 29.2 KB
external_callbacks.md -rw-r--r-- 5.5 KB
neural_network_with_tfds_data.ipynb -rw-r--r-- 14.3 KB
neural_network_with_tfds_data.md -rw-r--r-- 8.8 KB
quickstart.ipynb -rw-r--r-- 16.4 KB
quickstart.md -rw-r--r-- 9.8 KB
thinking_in_jax.ipynb -rw-r--r-- 79.2 KB
thinking_in_jax.md -rw-r--r-- 15.4 KB
vmapped_log_probs.ipynb -rw-r--r-- 41.0 KB
vmapped_log_probs.md -rw-r--r-- 6.3 KB
xmap_tutorial.ipynb -rw-r--r-- 422.1 KB
xmap_tutorial.md -rw-r--r-- 413.7 KB

README.md

back to top