Skip to content

Instantly share code, notes, and snippets.

Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@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 / exploring-kuramoto-sivashinsky-in-jax-cfd.ipynb
Last active October 14, 2021 23:52
Exploring Kuramoto-Sivashinsky in JAX-CFD.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-new-scan-benchmarking.ipynb
Last active August 29, 2021 23:28
JAX new scan benchmarking.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / new-gmres-benchmark.ipynb
Created December 1, 2020 21:03
new gmres benchmark.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@shoyer
shoyer / copy-of-benchmark_gmres.ipynb
Last active September 16, 2020 19:03
Copy of benchmark_gmres.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
# 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')