Search every repository hiding inside awesome lists.

Fast, flexible and easy to use probabilistic modelling in Python.

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

Accelerated deep learning R&D

Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

GeoAI: Artificial Intelligence for Geospatial Data

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

Data manipulation and transformation for audio signal processing, powered by PyTorch

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

PyTorch native quantization and sparsity for training and inference

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

Automatic architecture search and hyperparameter optimization for PyTorch

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

Machine learning metrics for distributed, scalable PyTorch applications.

An open source library and framework for deep learning on satellite and aerial imagery.

A toolkit for reproducible reinforcement learning research.

Automated Machine Learning on Kubernetes

An offline deep reinforcement learning library

Synthetic data generators for tabular and time-series data

A Data Streaming Library for Efficient Neural Network Training

OpenMMLab Foundational Library for Training Deep Learning Models

🛸 Use pretrained transformers like BERT, XLNet and GPT-2 in spaCy

🩹Editing large language models within 10 seconds⚡

Scalable and memory-optimized training of diffusion models

An open source DevOps tool from the CNCF for packaging and versioning AI/ML models, datasets, code, and configuration into an OCI Artifact.

code for deep learning courses

The good ol' Forge WebUI, now updated with new features~

Self-host the powerful Chatterbox TTS model. This server offers a user-friendly Web UI, flexible API endpoints (incl. OpenAI compatible), predefined voices, voice cloning, and large audiobook-scale text processing. Runs accelerated on NVIDIA (CUDA), AMD (ROCm), and CPU.

A JavaScript library like PyTorch, with GPU acceleration.

Transform geospatial relations into graphs for Graph Neural Networks and network analysis