Search awesome repositories

FAIR Sequence Modeling Toolkit 2

An autoML framework & toolkit for machine learning on graphs.

Audio processing by using pytorch 1D convolution network

DataDreamer: Prompt. Generate Synthetic Data. Train & Align Models.   🤖💤

Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

allRank is a framework for training learning-to-rank neural models based on PyTorch.

Pretrained model hub for Keras 3.

Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.

LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)

A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.

Build high-performance AI models with modular building blocks

TernausNetV2: Fully Convolutional Network for Instance Segmentation

Engine for AI/ML/Data tracking, visualization, explainability, drift detection, and dashboards for Polyaxon.

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

A Conversational Assistant equipped with synthetic voices including J.A.R.V.I.S's. Powered by OpenAI and IBM Watson APIs and a Tacotron model for voice generation.

A Libtorch implementation of the YOLO v3 object detection algorithm

Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

🦖 X—LLM: Cutting Edge & Easy LLM Finetuning

🔥TorchSat 🌏 is an open-source deep learning framework for satellite imagery analysis based on PyTorch.

tsl: a PyTorch library for processing spatiotemporal data.

PyTorch Implementation of OpenAI GPT-2

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

A full pipeline to finetune Vicuna LLM with LoRA and RLHF on consumer hardware. Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the Vicuna architecture. Basically ChatGPT but with Vicuna

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

eBPF agent and MCP server for GPU causal observability

Self-tuning multi-agent AI trading system. 8-source signal fusion, Bull/Bear/Judge debate on Claude Opus 4.7, Kelly + ATR position sizing. Python · Kalshi + Polymarket adapters.

RL stock selection for China A-share — bundled polars-native factor library (105 Alpha101 + 191 GTJA Alpha191 = 296 factors), board-aware price limits, GPU train + ONNX CPU infer, MIT-licensed.