Skip to content

Instantly share code, notes, and snippets.

View shoyer's full-sized avatar

Stephan Hoyer shoyer

973 followers · 23 following
View GitHub Profile
@shoyer
shoyer / jax-harmonic-oscillator-odeint.ipynb
Last active December 11, 2021 16:43
JAX harmonic oscillator odeint.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / contextvars_with_contextmanager.py
Last active August 16, 2020 00:29
# Copyright 2020 Google LLC.
# SPDX-License-Identifier: Apache-2.0
from contextvars import ContextVar
from contextlib import contextmanager
# library code
my_option = ContextVar('my_option', default='default')
@shoyer
shoyer / poisson-pytree-benchmark.ipynb
Created July 18, 2020 18:35
Poisson pytree benchmark
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 July 13, 2020 06:01
Simple JAX GMRES vectorized
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.ipynb
Created July 7, 2020 21:11
Simple JAX GMRES
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 June 16, 2020 06:42
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 June 8, 2020 20:57
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.
@shoyer
shoyer / JAX_TF_eager_autodiff_compatibility.ipynb
Last active August 6, 2022 15:04
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / jax-rng-class.ipynb
Created February 28, 2020 07:14
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 January 16, 2020 01:06
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.