Matthew Johnson
mattjj
research scientist @ Google Brain
Languages
Repos
81
Stars
2.5k
Forks
570
Top Language
Python
Loading contributions...
Top Repositories
Repositories
81A pedagogical implementation of Autograd
No description provided.
No description provided.
No description provided.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
No description provided.
Deep Learning for humans
some tools for gaussian linear dynamical systems
code for Structured Variational Autoencoders
Flax is a neural network library for JAX that is designed for flexibility.
my fish configuration
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
No description provided.
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
No description provided.
my config files, especially .tmux.conf
Multiple dispatch over abstract array types in JAX.
No description provided.
An Extensible Deep Learning Library
No description provided.
Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.
Fast C code for sampling Polya-gamma random variates. Builds on Jesse Windle's BayesLogit library.
No description provided.
No description provided.
beam sampling for pyhsmm
autoregressive plugin
👻
No description provided.
No description provided.
No description provided.