Stars
torchprime is a reference model implementation for PyTorch on TPU.
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.
A simple, performant and scalable Jax LLM!