Xinyang (Young) Geng
Xinyang (Young) Geng
CQL
Conservative Q Learning on top of SAC
JaxCQL
Conservative Q learning in Jax
m3ae_public
Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation
EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
mlxu
Machine Learning eXperiment Utilities
scalax
A simple library for scaling up JAX programs
koala_data_pipeline
The data processing pipeline for the Koala chatbot language model
mintext
Minimal but scalable implementation of large language models in JAX
SimpleSAC
A simple and easy to use implementation of the soft actor-critic algorithm.