young-geng
Follow
-
orbax Public
Forked from google/orbaxOrbax provides common checkpointing and persistence utilities for JAX users
-
gpu_kernel_excercise Public
Learning to write high performance GPU kernels
Python MIT License UpdatedDec 12, 2025 -
mintext Public
Minimal but scalable implementation of large language models in JAX
-
scalax Public
A simple library for scaling up JAX programs
-
tpu_pod_commander Public
TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
-
cuda-training-series Public
Forked from olcf/cuda-training-seriesTraining materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)
-
mlxu Public
Machine Learning eXperiment Utilities
-
bagz Public
Forked from google-deepmind/bagzBagz is a format for storing a sequence of string records. It supports per-record compression and fast index-based lookup.
C++ Apache License 2.0 UpdatedJul 22, 2025 -
-
m3ae_public Public
Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation
-
-
EasyLM Public
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
-
-
koala_data_pipeline Public
The data processing pipeline for the Koala chatbot language model
-
bairblog.github.io Public
Forked from bairblog/bairblog.github.ioJavaScript MIT License UpdatedApr 4, 2023 -
FastChat Public
Forked from lm-sys/FastChatThe release repo for "Vicuna: An Open Chatbot Impressing GPT-4"
Python Apache License 2.0 UpdatedApr 3, 2023 -
mmlu_easylm Public
Forked from hendrycks/testMMLU evaluation for language models served with EasyLM
-
transformers Public
Forked from huggingface/transformers🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Python Apache License 2.0 UpdatedMar 2, 2023 -
JaxCQL Public
Conservative Q learning in Jax
-
CQL Public
Conservative Q Learning on top of SAC
-
jax-fid Public
Forked from matthias-wright/jax-fidFID computation in Jax/Flax.
Python Apache License 2.0 UpdatedSep 26, 2022 -
SimpleSAC Public
A simple and easy to use implementation of the soft actor-critic algorithm.
-
-
-
-
design-bench Public
Forked from brandontrabucco/design-benchBenchmarks for Model-Based Optimization
Python MIT License UpdatedJan 5, 2022 -
dopamine Public
Forked from google/dopamineDopamine is a research framework for fast prototyping of reinforcement learning algorithms.
-
d4rl Public
Forked from Farama-Foundation/D4RLA benchmark for offline reinforcement learning.
-
-
parameterized_model Public
TensorFlow parameterized model library