Give AlbumentationsX a star on GitHub — it powers this leaderboard
Minimal library to train LLMs on TPU in JAX with pjit().