Pinned Loading

1. golf golf Public

   Gradient optimized breakpoint finding for piecewise linear fitting

   Python 2

2. equintk equintk Public

   A JAX+EQUINOX implementation of NTK and NNGP calculations as https://github.com/google/neural-tangents was archived while still buggy

   Jupyter Notebook 1

3. INR_BEP INR_BEP Public

   Forked from SimonKoop/inr_edu_24

   Fork for the bep

   Jupyter Notebook

4. stacy stacy Public

   my nix config modularized with flakes

   Nix 1

5. equimixedp equimixedp Public

   Mixed precision training with equinox

   Python 1

6. pallas-jax-flash-attention pallas-jax-flash-attention Public

   end goal is to write a jax kernel for flash attention

   Python