Skip to content

Instantly share code, notes, and snippets.

@kazewong
kazewong / spline.py
Last active December 2, 2023 16:55
Cubic Spline with lineax
import lineax as lx
import jax.numpy as jnp
import jax
from jaxtyping import Float, Array
class CubicSpline:
x_grid: Float[Array, str("batch")] # input x data
y_grid: Float[Array, str("n")] # input y data
@kazewong
kazewong / GW150914.ipynb
Created October 6, 2022 03:05
Notebook for running GW150914 with FlowMC
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@kazewong
kazewong / IntroductionToFlax.ipynb
Last active June 29, 2022 18:22
Introduction to flax
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.