Pinned Repositories
maxtext
A simple, performant and scalable Jax LLM!
xpk
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.
ml-testing-accelerators
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
accelerate
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
hf-transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
xpk
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.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
jonb377's Repositories
jonb377/accelerate
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
jonb377/hf-transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
jonb377/nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
jonb377/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
jonb377/xpk
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.