Skip to content

Instantly share code, notes, and snippets.

@shoyer
shoyer / jaxopt-implicit-optimization-gradients-with-jax-md.ipynb
Created Feb 8, 2022
JAXopt implicit optimization gradients with JAX-MD.ipynb
View jaxopt-implicit-optimization-gradients-with-jax-md.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 Oct 14, 2021
Exploring Kuramoto-Sivashinsky in JAX-CFD.ipynb
View 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 Aug 29, 2021
JAX new scan benchmarking.ipynb
View 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 Dec 1, 2020
new gmres benchmark.ipynb
View 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.
View fast_groupby_bins_agg.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View dask-memory-bug.ipynb
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 Sep 16, 2020
Copy of benchmark_gmres.ipynb
View 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.
@shoyer
shoyer / jax-harmonic-oscillator-odeint.ipynb
Last active Dec 11, 2021
JAX harmonic oscillator odeint.ipynb
View 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.
View contextvars_with_contextmanager.py
# 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')
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.