AI Agent Skills

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

This project reproduces the book Dive Into Deep Learning (https://d2l.ai/), adapting the code from MXNet into PyTorch.

A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.

Code Repository for Machine Learning with PyTorch and Scikit-Learn

Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks…

Simplest distributed training API. 4 lines to add distributed support to any PyTorch script. Unified API for DeepSpeed/FSDP/Megatron/DDP. Automatic device placement, mix…

Lightning-fast and Powerful Code Editor written in Rust

Open Lakehouse Format for Multimodal AI. Convert from Parquet in 2 lines of code for 100x faster random access, vector index, and data versioning. Compatible with Pandas…

Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).