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.
xuefgu doesn’t have any repository yet.