Xinyang (Young) Geng

Results 6 repositories owned by Xinyang (Young) Geng

CQL

112
Stars
23
Forks
Watchers

Conservative Q Learning on top of SAC

JaxCQL

47
Stars
6
Forks
Watchers

Conservative Q learning in Jax

m3ae_public

93
Stars
12
Forks
Watchers

Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation

EasyLM

2.3k
Stars
233
Forks
12
Watchers

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

mlxu

37
Stars
6
Forks
Watchers

Machine Learning eXperiment Utilities

scalax

106
Stars
6
Forks
Watchers

A simple library for scaling up JAX programs