A JAX-native LLM Post-Training Library - View it on GitHub