Tunix

A lightweight, JAX-native library for LLM post-training — SFT, RLHF, DPO, and more.