Roy Frostig

Coauthor of JAX; Research Scientist at Google DeepMind

Starred Projects (25)

ml-engineering by stas00

Open book for LLM/VLM training engineers

tobi:

jrk:

casper-hansen:

willccbb:

Created 5 years ago

Updated 1 day ago

marin by marin-community

Framework for reproducible foundation model research and development

john-b-yang:

percyliang:

hammer:

teetone:

Created 1 year ago

Updated 19 hours ago

scaling-book by jax-ml

LLM scaling guide on TPUs

srush:

ekzhang:

mattjj:

Created 7 months ago

Updated 4 days ago

openpi by Physical-Intelligence

Robotics vision-language-action models

hammer:

forresti:

codekansas:

sxyu:

Created 11 months ago

Updated 1 week ago

prompt-tuning-playbook by varungodbole

Prompt engineering guide for post-trained LLMs

Created 10 months ago

Updated 8 months ago

entropix by xjdr-alt

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

geohot:

vincentweisser:

snowzurfer:

evhub:

Created 11 months ago

Updated 10 months ago

llm by simonw

CLI tool and Python library for LLM interaction

hugs:

willccbb:

binarybana:

willingc:

Created 2 years ago

Updated 1 month ago

HighPerfLLMs2024 by rwitten

Jax course for high-performance LLM construction

Jiayi-Pan:

hammer:

mattjj:

Created 1 year ago

Updated 1 year ago

JetStream by AI-Hypercomputer

LLM inference engine optimized for throughput and memory on XLA devices

osanseviero:

hammer:

Created 1 year ago

Updated 3 months ago

einx by fferflo

Tensor operation library using Einstein-inspired notation

awni:

patrick-kidger:

jiamings:

hammer:

Created 2 years ago

Updated 5 months ago

penzai by google-deepmind

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

chiphuyen:

nikiparmar:

Edward-Sun:

kelvich:

Created 1 year ago

Updated 2 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:

quincylarson:

Created 2 years ago

Updated 7 months ago

vllm by vllm-project

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

karpathy:

clmnt:

tobi:

danielhanchen:

Created 2 years ago

Updated 14 hours ago

sglang by sgl-project

Fast serving framework for LLMs and vision language models

samlambert:

ebursztein:

philschmid:

pankajroark:

Created 1 year ago

Updated 14 hours ago

diffrax by patrick-kidger

JAX library for numerical differential equation solvers

patrickvonplaten:

lucidrains:

patrick-kidger:

Created 4 years ago

Updated 6 days ago

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

omarsar:

yang-song:

hammer:

ibab:

Created 6 years ago

Updated 1 week ago

mlx by ml-explore

Array framework for machine learning on Apple silicon

geohot:

karpathy:

jmorganca:

chiphuyen:

Created 1 year ago

Updated 2 days ago

dspy by stanfordnlp

Framework for programming language models, not prompting

tobi:

mateiz:

vincentweisser:

pgarbacki:

Created 2 years ago

Updated 15 hours ago

maxtext by AI-Hypercomputer

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

sebastiandero:

Edward-Sun:

jph00:

transitive-bullshit:

Created 2 years ago

Updated 17 hours ago

levanter by marin-community

Framework for training foundation models with JAX

pcmoritz:

Jiayi-Pan:

thomwolf:

chiphuyen:

Created 3 years ago

Updated 18 hours 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

jaxformer by salesforce

JAX library for LLM training on TPUs

hammer:

Edward-Sun:

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

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

Feedback? Help us improve.