JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning. From Wikipedia
The new 9B and 27B parameter models offer high performance and efficiency, accessible through various AI frameworks.