Stars
A simple, performant and scalable Jax LLM!
xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.
Package of Pathways-on-Cloud utilities
lukebaumann / axlearn
Forked from apple/axlearnAn Extensible Deep Learning Library