GitHub repository: ADAPT-uiuc/sabertooth
Primary language: Python
Description: Standalone pre-training recipe with JAX+Flax