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