Matthew Johnson

Coauthor of JAX; Research Scientist at Google Brain

Starred Projects (53)

jax-js by ekzhang

JAX-style ML framework for the web

luiscape:

srush:

abhiaiyer91:

smthomas:

Created 1 year ago

Updated 21 hours ago

sglang-jax by sgl-project

High-performance LLM inference engine for JAX/TPU serving

pcmoritz:

froystig:

hammer:

merrymercy:

Created 1 year ago

Updated 3 days ago

jax-llm-examples by jax-ml

High-performance LLM implementations in JAX

codekansas:

hammer:

froystig:

Created 1 year ago

Updated 3 weeks ago

scaling-book by jax-ml

LLM scaling guide on TPUs

codekansas:

srush:

froystig:

ekzhang:

Created 1 year ago

Updated 21 hours ago

felafax by felafax

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

jrk:

luiscape:

hammer:

deshraj:

Created 2 years ago

Updated 1 year ago

jax-ai-stack by jax-ml

A curated JAX ecosystem for generative AI development

hammer:

Created 2 years ago

Updated 1 week ago

EasyDeL by erfanzar

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

hammer:

Jiayi-Pan:

Created 3 years ago

Updated 4 days ago

maxdiffusion by AI-Hypercomputer

Jax diffusion models for training and inference

hammer:

Created 2 years ago

Updated 2 days ago

HighPerfLLMs2024 by rwitten

Jax course for high-performance LLM construction

froystig:

Jiayi-Pan:

hammer:

Created 2 years ago

Updated 2 years ago

cutlass by NVIDIA

CUDA C++ and Python DSLs for high-performance linear algebra

luiscape:

pgarbacki:

pankajroark:

tridao:

Created 8 years ago

Updated 6 days ago

equinox by patrick-kidger

JAX library for neural networks and scientific computing

hammer:

jrk:

hanlint:

patrick-kidger:

Created 5 years ago

Updated 4 weeks ago

EasyLM by young-geng

LLM training/finetuning framework in JAX/Flax

Jiayi-Pan:

chiphuyen:

shizhediao:

zhuohan123:

Created 3 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 3 years ago

Updated 2 years ago

jax-triton by jax-ml

JAX integrations for OpenAI Triton

hammer:

gakonst:

ogabrielluiz:

patrickvonplaten:

Created 4 years ago

Updated 4 days ago

iree by iree-org

MLIR-based compiler and runtime toolkit for machine learning models

zhyncs:

ogabrielluiz:

mfuntowicz:

parano:

Created 6 years ago

Updated 3 days ago

dalle-mini by borisdayma

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

karpathy:

hardmaru:

infwinston:

codekansas:

Created 5 years ago

Updated 2 years ago

awesome-jax by n2cholas

Curated list of JAX resources for high-performance ML research

aravindsrinivas:

hammer:

dynamicwebpaige:

patrickvonplaten:

Created 5 years ago

Updated 6 months ago

kotlingrad by breandan

Symbolic differentiation framework for the JVM

hammer:

Created 7 years ago

Updated 1 year 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 6 years ago

Updated 5 days ago

scalene by plasma-umass

Python profiler with AI-powered optimization proposals

luiscape:

xiezhq-hermann:

zhuohan123:

Ying1123:

Created 6 years ago

Updated 3 weeks ago

tsalib by ofnote

Tensor Shape Annotation Library (tsalib) enables named tensor dimensions

jekbradbury:

apaszke:

Created 7 years ago

Updated 6 years ago

pyhf by scikit-hep

HistFactory implementation for fitting/limit-setting/interval estimation

lantiga:

soumith:

hammer:

Created 8 years ago

Updated 6 days ago

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

omarsar:

froystig:

yang-song:

hammer:

Created 7 years ago

Updated 6 days ago

ray by ray-project

AI compute engine for scaling Python and AI applications

didierrlopes:

beyang:

hsbt:

gregpr07:

Created 9 years ago

Updated 14 hours ago

ebisu by fasiha

Bayesian spaced repetition algorithm for flashcard quiz scheduling

Created 9 years ago

Updated 1 year ago

tensorlang by tensorlang

Differentiable programming for TensorFlow

Created 9 years ago

Updated 8 years ago

multiagent-particle-envs by openai

Multi-agent environment for actor-critic research

aravindsrinivas:

evhub:

apsdehal:

AntonOsika:

Created 9 years ago

Updated 2 years ago

horovod by horovod

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

aravindsrinivas:

fchollet:

winglian:

transitive-bullshit:

Created 9 years ago

Updated 1 month ago

robosumo by openai

Code for research paper on meta-learning in competitive environments

aravindsrinivas:

jinze1994:

jiamings:

Created 8 years ago

Updated 3 years ago

mujoco-py by openai

Python bindings for MuJoCo physics engine

lilianweng:

aravindsrinivas:

ogabrielluiz:

evhub:

Created 10 years ago

Updated 2 years ago

baselines by openai

RL algorithm implementations for research

aravindsrinivas:

lilianweng:

JohannesHa:

jspahrsummers:

Created 9 years ago

Updated 2 years ago

openai-gemm by openai

GEMM kernels for single/half precision

jrk:

jiamings:

s9xie:

deshraj:

Created 9 years ago

Updated 3 years ago

ganhacks by soumith

GAN training tips and tricks

aravindsrinivas:

lilianweng:

ebursztein:

stefanv:

Created 9 years ago

Updated 4 years ago

pixel-cnn by openai

TensorFlow implementation for PixelCNN++ research paper

aravindsrinivas:

evhub:

jinze1994:

omarsar:

Created 10 years ago

Updated 6 years ago

reinforcement-learning by dennybritz

RL algorithms implementation for education

lilianweng:

ogabrielluiz:

thomwolf:

jinze1994:

Created 10 years ago

Updated 3 years ago

cleverhans by cleverhans-lab

Adversarial example library for benchmarking ML model robustness

lilianweng:

huybery:

generall:

colah:

Created 10 years ago

Updated 2 years ago

awesome-tensorflow by jtoy

Curated list of TensorFlow resources

lilianweng:

osanseviero:

antiagainst:

vincentweisser:

Created 10 years ago

Updated 5 months ago

InfoGAN by openai

Research paper implementation for interpretable representation learning via InfoGAN

aravindsrinivas:

evhub:

ericjang:

lukemetz:

Created 10 years ago

Updated 5 years ago

async-rl by coreylynch

TensorFlow/Keras implementation of async RL research paper

aravindsrinivas:

fchollet:

agermanidis:

junxiaosong:

Created 10 years ago

Updated 8 years ago

improved-gan by openai

GAN training code for research paper replication

aravindsrinivas:

lilianweng:

bcherny:

evhub:

Created 10 years ago

Updated 7 years ago

compromise by spencermountain

JavaScript NLP toolkit for text data transformation

bcherny:

mikermcneil:

matamalaortiz:

steve8708:

Created 15 years ago

Updated 1 week ago

gym by openai

RL toolkit for reinforcement learning algorithm development

aravindsrinivas:

fchollet:

bcherny:

didierrlopes:

Created 10 years ago

Updated 4 months ago

rllab by rll

Framework for reinforcement learning algorithm development and evaluation

aravindsrinivas:

gdb:

apsdehal:

parano:

Created 10 years ago

Updated 3 years ago

spaCy by explosion

NLP library for production applications

aravindsrinivas:

fchollet:

nirga:

jn2clark:

Created 12 years ago

Updated 2 months ago

CNTK by microsoft

Deep-learning toolkit for neural networks as directed computational graphs

bcherny:

tiangolo:

luiscape:

parano:

Created 10 years ago

Updated 3 years ago

tqdm by tqdm

Progress bar for Python and CLI tools

bcherny:

geohot:

hugs:

parano:

Created 11 years ago

Updated 2 days 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 17 years ago

Updated 3 days ago

neon by NervanaSystems

Deep learning framework (discontinued)

fchollet:

parano:

trishume:

hanlint:

Created 11 years ago

Updated 5 years ago

cloudpickle by cloudpipe

Extended pickling support for Python objects

borzunov:

youkaichao:

luiscape:

hammer:

Created 11 years ago

Updated 8 months ago

OpenBLAS by OpenMathLib

Optimized BLAS library for linear algebra subprograms

kerollmops:

jiamings:

infwinston:

jcjohnson:

Created 15 years ago

Updated 16 hours ago

portaljs by datopian

Data portal builder framework

luiscape:

Created 15 years ago

Updated 1 week ago

Feedback? Help us improve.