Skip to content

Instantly share code, notes, and snippets.

View poisson-pytree-benchmark.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / simple-jax-gmres-vectorized.ipynb
Last active Jul 13, 2020
Simple JAX GMRES vectorized
View simple-jax-gmres-vectorized.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View simple-jax-gmres.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / numba-vs-jax-filter.ipynb
Last active Jun 16, 2020
numba vs jax filter.ipynb
View numba-vs-jax-filter.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / jax-einsum-primitive.ipynb
Last active Jun 8, 2020
JAX einsum primitive .ipynb
View jax-einsum-primitive.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View JAX_TF_eager_autodiff_compatibility.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View jax-rng-class.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / jax-odeint-benchmark.ipynb
Created Jan 16, 2020
jax odeint benchmark.ipynb
View jax-odeint-benchmark.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / jax-gradient-benchmark.ipynb
Created Jan 8, 2020
jax gradient benchmark.ipynb
View jax-gradient-benchmark.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / numpy-duck-array-ecosystem-graph.ipynb
Created Aug 28, 2019
NumPY duck array ecosystem graph
View numpy-duck-array-ecosystem-graph.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
You can’t perform that action at this time.