Roy Frostig

Coauthor of JAX; Research Scientist at Google DeepMind

Starred Projects (27)

sglang-jax by sgl-project

High-performance LLM inference engine for JAX/TPU serving

mattjj:

hammer:

merrymercy:

zhyncs:

Created 10 months ago

Updated 9 hours ago

ml-engineering by stas00

Open book for LLM/VLM training engineers

tobi:

WoosukKwon:

ppwwyyxx:

jrk:

Created 5 years ago

Updated 1 week ago

marin by marin-community

Framework for reproducible foundation model research and development

natolambert:

lysandrejik:

john-b-yang:

percyliang:

Created 2 years ago

Updated 9 hours ago

jax-llm-examples by jax-ml

High-performance LLM implementations in JAX

codekansas:

hammer:

mattjj:

Created 1 year ago

Updated 1 month ago

scaling-book by jax-ml

LLM scaling guide on TPUs

codekansas:

srush:

ekzhang:

mattjj:

Created 1 year ago

Updated 4 days ago

openpi by Physical-Intelligence

Robotics vision-language-action models

brianichter:

hammer:

forresti:

codekansas:

Created 1 year ago

Updated 3 weeks ago

prompt-tuning-playbook by varungodbole

Prompt engineering guide for post-trained LLMs

Created 1 year ago

Updated 1 year ago

entropix by xjdr-alt

Research project for entropy-based context-aware sampling & parallel CoT decoding

geohot:

vincentweisser:

snowzurfer:

evhub:

Created 1 year ago

Updated 1 year ago

llm by simonw

CLI tool and Python library for LLM interaction

hugs:

willccbb:

binarybana:

willingc:

Created 3 years ago

Updated 1 week ago

HighPerfLLMs2024 by rwitten

Jax course for high-performance LLM construction

Jiayi-Pan:

hammer:

mattjj:

Created 2 years ago

Updated 1 year ago

JetStream by AI-Hypercomputer

LLM inference engine optimized for throughput and memory on XLA devices

osanseviero:

hammer:

Created 2 years ago

Updated 4 months ago

einx by fferflo

Tensor operation library using Einstein-inspired notation

awni:

patrick-kidger:

jiamings:

hammer:

Created 2 years ago

Updated 3 weeks ago

penzai by google-deepmind

JAX research toolkit for neural network building, editing, and visualization

chiphuyen:

nikiparmar:

Edward-Sun:

kelvich:

Created 2 years ago

Updated 11 months ago

nanodl by HenryNdubuaku

Jax library for building transformer models, including GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT

hammer:

Created 2 years ago

Updated 1 year ago

llama by meta-llama

Inference code for Llama 2 models (deprecated)

xiezhq-hermann:

fabhed:

borzunov:

calebpeffer:

Created 3 years ago

Updated 1 year ago

vllm by vllm-project

LLM serving engine for high-throughput, memory-efficient inference

karpathy:

clmnt:

tobi:

adamdotdevin:

Created 3 years ago

Updated 9 hours ago

sglang by sgl-project

Fast serving framework for LLMs and vision language models

amin3141:

shimmyshimmer:

beyang:

samlambert:

Created 2 years ago

Updated 9 hours ago

diffrax by patrick-kidger

JAX library for numerical differential equation solvers

patrickvonplaten:

lucidrains:

patrick-kidger:

Created 4 years ago

Updated 1 week ago

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

omarsar:

yang-song:

hammer:

ibab:

Created 7 years ago

Updated 2 days ago

mlx by ml-explore

Array framework for machine learning on Apple silicon

geohot:

karpathy:

ekzhang:

lysandrejik:

Created 2 years ago

Updated 3 days ago

dspy by stanfordnlp

Framework for programming language models, not prompting

tobi:

mateiz:

vincentweisser:

pgarbacki:

Created 3 years ago

Updated 11 hours ago

maxtext by AI-Hypercomputer

Jax LLM for high-performance, scalable training/inference on TPUs/GPUs

ebursztein:

sebastiandero:

Edward-Sun:

jph00:

Created 3 years ago

Updated 12 hours ago

levanter by marin-community

Framework for training foundation models with JAX

pcmoritz:

Jiayi-Pan:

thomwolf:

chiphuyen:

Created 4 years ago

Updated 4 months 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

jaxformer by salesforce

JAX library for LLM training on TPUs

hammer:

Edward-Sun:

Created 3 years ago

Updated 2 years ago

jax-triton by jax-ml

JAX integrations for OpenAI Triton

hammer:

gakonst:

ogabrielluiz:

patrickvonplaten:

Created 3 years ago

Updated 1 month 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

Feedback? Help us improve.