Pinned Repositories
autodidact
A pedagogical implementation of Autograd
my-oh-my-zsh
pybasicbayes
pyhsmm
pyhsmm-autoregressive
autoregressive plugin
pyhsmm-factorial
pylds
some tools for gaussian linear dynamical systems
pyslds
svae
code for Structured Variational Autoencoders
mattjj's Repositories
mattjj/autodidact
A pedagogical implementation of Autograd
mattjj/pyhsmm
mattjj/svae
code for Structured Variational Autoencoders
mattjj/pybasicbayes
mattjj/pyslds
mattjj/pylds
some tools for gaussian linear dynamical systems
mattjj/pyhsmm-autoregressive
autoregressive plugin
mattjj/ode-diff-notes
mattjj/variational_autoencoder
mattjj/pplham
mattjj/spectral_clustering
mattjj/oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
mattjj/config-vim
my .vim
mattjj/kalman_grads
mattjj/autograd_linalg
improve scipy.linalg and corresponding autograd gradfuns
mattjj/computation-thru-dynamics
Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.
mattjj/axlearn
An Extensible Deep Learning Library
mattjj/dex-lang
Research language for array processing in the Haskell/ML family
mattjj/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
mattjj/pypolyagamma
Fast C code for sampling Polya-gamma random variates. Builds on Jesse Windle's BayesLogit library.
mattjj/flax
Flax is a neural network library for JAX that is designed for flexibility.
mattjj/jax-md
Differentiable, Hardware Accelerated, Molecular Dynamics
mattjj/blog-benchmark-rwmetropolis
Code to reproduce the results of https://rlouf.github.io/post/jax-random-walk-metropolis/
mattjj/flash-attention-jax
Implementation of Flash Attention in Jax
mattjj/HighPerfLLMs2024
mattjj/iree
👻
mattjj/jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
mattjj/julia
The Julia Language: A fresh approach to technical computing.
mattjj/mattjj.github.io
mattjj/mpi4py-examples
mpi4py examples