Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

A puzzle to learn about prompting

What would you do with 1000 H100s...

Serve, optimize and scale PyTorch models in production

LlamaIndex is a data framework for your LLM applications

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

PyMC educational resources

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.