Programming language for ML on XLA with JAX-level speed; compiles for CPU, TPUs/Nvidia GPUs, AMD GPUs, Apple GPUs, etc. - View it on GitHub