Fine-tuning & Reinforcement Learning for LLMs. 🦥 Train OpenAI gpt-oss, DeepSeek-R1, Qwen3, Gemma 3, TTS 2x faster with 70% less VRAM.

The definitive Web UI for local AI, with powerful features and easy setup.

Expose your FastAPI endpoints as Model Context Protocol (MCP) tools, with Auth!

OneTrainer is a one-stop solution for all your stable diffusion training needs.

Feature engineering package with sklearn like functionality

A Python toolkit/library for reality-centric machine/deep learning and data mining on partially-observed time series, including SOTA neural network models for scientific analysis tasks of imputatio…

A library to inspect and extract intermediate layers of PyTorch models.

Augmentation pipeline for rendering synthetic paper printing, faxing, scanning and copy machine processes

Deploy BERT for Sentiment Analysis as REST API using FastAPI, Transformers by Hugging Face and PyTorch

Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels

PyTorch – SMART: Robust and Efficient Fine-Tuning for Pre-trained Natural Language Models.

Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719

Virtual Adversarial Training (VAT) techniques in PyTorch