Xinyang (Young) Geng

Results 11 repositories owned by Xinyang (Young) Geng

CQL

123
Stars
25
Forks
Watchers

Conservative Q Learning on top of SAC

JaxCQL

52
Stars
6
Forks
Watchers

Conservative Q learning in Jax

m3ae_public

101
Stars
12
Forks
Watchers

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

EasyLM

2.5k
Stars
260
Forks
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

46
Stars
8
Forks
Watchers

Machine Learning eXperiment Utilities

scalax

129
Stars
10
Forks
Watchers

A simple library for scaling up JAX programs

gps_superball_public

16
Stars
7
Forks
Watchers

GPS codebase for NASA Superball tensegrity robot

koala_data_pipeline

117
Stars
13
Forks
Watchers

The data processing pipeline for the Koala chatbot language model

mintext

32
Stars
0
Forks
Watchers

Minimal but scalable implementation of large language models in JAX

SimpleSAC

15
Stars
1
Forks
Watchers

A simple and easy to use implementation of the soft actor-critic algorithm.