Matthew Johnson

Coauthor of JAX; Research Scientist at Google Brain

Starred Projects (46)

scaling-book by jax-ml

LLM scaling guide on TPUs

srush:

froystig:

ekzhang:

Created 7 months ago

Updated 4 days ago

felafax by felafax

AI infra for non-NVIDIA GPUs, enabling LLM fine-tuning

jrk:

luiscape:

hammer:

deshraj:

Created 1 year ago

Updated 7 months ago

EasyDeL by erfanzar

Framework for streamlined JAX model training/serving on TPU/GPU

hammer:

Jiayi-Pan:

Created 2 years ago

Updated 1 day ago

maxdiffusion by AI-Hypercomputer

Jax diffusion models for training and inference

Created 2 years ago

Updated 1 day ago

HighPerfLLMs2024 by rwitten

Jax course for high-performance LLM construction

froystig:

Jiayi-Pan:

hammer:

Created 1 year ago

Updated 1 year ago

equinox by patrick-kidger

JAX library for neural networks and scientific computing

hammer:

jrk:

hanlint:

patrick-kidger:

Created 4 years ago

Updated 3 weeks ago

EasyLM by young-geng

LLM training/finetuning framework in JAX/Flax

Jiayi-Pan:

chiphuyen:

shizhediao:

zhuohan123:

Created 2 years ago

Updated 1 year ago

whisper-jax by sanchit-gandhi

JAX implementation of OpenAI's Whisper model for fast TPU inference

tobi:

patrickvonplaten:

parano:

osanseviero:

Created 2 years ago

Updated 1 year ago

jax-triton by jax-ml

JAX integrations for OpenAI Triton

hammer:

gakonst:

ogabrielluiz:

patrickvonplaten:

Created 3 years ago

Updated 2 weeks ago

iree by iree-org

MLIR-based compiler and runtime toolkit for machine learning models

ogabrielluiz:

mfuntowicz:

parano:

infwinston:

Created 6 years ago

Updated 16 hours ago

dalle-mini by borisdayma

Text-to-image model for generating images from text prompts

karpathy:

hardmaru:

infwinston:

codekansas:

Created 4 years ago

Updated 1 year ago

awesome-jax by n2cholas

Curated list of JAX resources for high-performance ML research

aravindsrinivas:

hammer:

dynamicwebpaige:

patrickvonplaten:

Created 4 years ago

Updated 2 weeks ago

kotlingrad by breandan

Symbolic differentiation framework for the JVM

hammer:

Created 7 years ago

Updated 9 months ago

whynot by socialfoundations

Python sandbox for decision making in dynamics

hammer:

MishaLaskin:

Created 6 years ago

Updated 2 years ago

flax by google

NN library for JAX, designed for flexibility in neural network research

charliermarsh:

codekansas:

Jiayi-Pan:

jaredpalmer:

Created 5 years ago

Updated 19 hours ago

scalene by plasma-umass

Python profiler with AI-powered optimization proposals

luiscape:

xiezhq-hermann:

zhuohan123:

Ying1123:

Created 5 years ago

Updated 1 week ago

tsalib by ofnote

Tensor Shape Annotation Library (tsalib) enables named tensor dimensions

jekbradbury:

apaszke:

Created 7 years ago

Updated 5 years ago

pyhf by scikit-hep

HistFactory implementation for fitting/limit-setting/interval estimation

lantiga:

soumith:

hammer:

Created 7 years ago

Updated 1 week ago

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

omarsar:

froystig:

yang-song:

hammer:

Created 6 years ago

Updated 1 week ago

ray by ray-project

AI compute engine for scaling Python and AI applications

beyang:

hsbt:

gregpr07:

hiyouga:

Created 9 years ago

Updated 14 hours ago

ebisu by fasiha

Bayesian spaced repetition algorithm for flashcard quiz scheduling

Created 8 years ago

Updated 11 months ago

multiagent-particle-envs by openai

Multi-agent environment for actor-critic research

aravindsrinivas:

evhub:

apsdehal:

AntonOsika:

Created 8 years ago

Updated 1 year ago

horovod by horovod

Distributed training framework for TF, Keras, PyTorch, and MXNet

aravindsrinivas:

fchollet:

winglian:

transitive-bullshit:

Created 8 years ago

Updated 2 days ago

robosumo by openai

Code for research paper on meta-learning in competitive environments

aravindsrinivas:

jinze1994:

jiamings:

Created 8 years ago

Updated 2 years ago

mujoco-py by openai

Python bindings for MuJoCo physics engine

lilianweng:

aravindsrinivas:

ogabrielluiz:

evhub:

Created 9 years ago

Updated 1 year ago

baselines by openai

RL algorithm implementations for research

aravindsrinivas:

lilianweng:

JohannesHa:

jspahrsummers:

Created 8 years ago

Updated 1 year ago

openai-gemm by openai

GEMM kernels for single/half precision

jrk:

jiamings:

s9xie:

deshraj:

Created 9 years ago

Updated 2 years ago

ganhacks by soumith

GAN training tips and tricks

aravindsrinivas:

lilianweng:

ebursztein:

stefanv:

Created 8 years ago

Updated 3 years ago

pixel-cnn by openai

TensorFlow implementation for PixelCNN++ research paper

aravindsrinivas:

evhub:

jinze1994:

omarsar:

Created 9 years ago

Updated 5 years ago

reinforcement-learning by dennybritz

RL algorithms implementation for education

lilianweng:

ogabrielluiz:

thomwolf:

jinze1994:

Created 9 years ago

Updated 2 years ago

cleverhans by cleverhans-lab

Adversarial example library for benchmarking ML model robustness

lilianweng:

huybery:

generall:

colah:

Created 9 years ago

Updated 1 year ago

awesome-tensorflow by jtoy

Curated list of TensorFlow resources

lilianweng:

osanseviero:

antiagainst:

vincentweisser:

Created 9 years ago

Updated 6 months ago

InfoGAN by openai

Research paper implementation for interpretable representation learning via InfoGAN

aravindsrinivas:

evhub:

ericjang:

lukemetz:

Created 9 years ago

Updated 4 years ago

async-rl by coreylynch

TensorFlow/Keras implementation of async RL research paper

aravindsrinivas:

fchollet:

agermanidis:

junxiaosong:

Created 9 years ago

Updated 7 years ago

improved-gan by openai

GAN training code for research paper replication

aravindsrinivas:

lilianweng:

bcherny:

evhub:

Created 9 years ago

Updated 6 years ago

gym by openai

RL toolkit for reinforcement learning algorithm development

aravindsrinivas:

fchollet:

bcherny:

ValentaTomas:

Created 9 years ago

Updated 11 months ago

rllab by rll

Framework for reinforcement learning algorithm development and evaluation

aravindsrinivas:

gdb:

apsdehal:

parano:

Created 9 years ago

Updated 2 years ago

spaCy by explosion

NLP library for production applications

aravindsrinivas:

fchollet:

nirga:

jn2clark:

Created 11 years ago

Updated 3 months ago

stat212b by joanbruna

DL course material (UC Berkeley, Spring 2016)

AntonOsika:

adityaramesh:

lantiga:

lukemetz:

Created 9 years ago

Updated 8 years ago

CNTK by microsoft

Deep-learning toolkit for neural networks as directed computational graphs

bcherny:

tiangolo:

luiscape:

parano:

Created 9 years ago

Updated 2 years ago

tqdm by tqdm

Progress bar for Python and CLI tools

bcherny:

geohot:

hugs:

parano:

Created 10 years ago

Updated 3 months ago

tensorflow by tensorflow

Open-source ML framework

norvig:

aravindsrinivas:

karpathy:

bcherny:

Created 10 years ago

Updated 14 hours ago

redis by redis

Redis is a versatile data structure server, cache, and query engine

mikeyk:

patrickvonplaten:

nirga:

chiphuyen:

Created 16 years ago

Updated 3 days ago

neon by NervanaSystems

Deep learning framework (discontinued)

fchollet:

parano:

trishume:

hanlint:

Created 11 years ago

Updated 4 years ago

cloudpickle by cloudpipe

Extended pickling support for Python objects

borzunov:

youkaichao:

luiscape:

hammer:

Created 10 years ago

Updated 3 weeks ago

OpenBLAS by OpenMathLib

Optimized BLAS library for linear algebra subprograms

kerollmops:

jiamings:

infwinston:

jcjohnson:

Created 14 years ago

Updated 1 day ago

Feedback? Help us improve.