All repos — GH Trending Explorer

pytorch / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

keras-team / keras

Deep Learning for humans

explosion / spaCy

💫 Industrial-strength Natural Language Processing (NLP) in Python

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch

onnx / onnx

Open standard for machine learning interoperability

NVIDIA-NeMo / Speech

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

ludwig-ai / ludwig

Low-code framework for building custom LLMs, neural networks, and other AI models

kornia / kornia

🐍 Geometric Computer Vision Library for Spatial AI

rushter / MLAlgorithms

Minimal and clean examples of machine learning algorithms implementations

zyddnys / manga-image-translator

Translate manga/image 一键翻译各类图片内文字 https://cotrans.touhou.ai/ (no longer working)

google-deepmind / sonnet

TensorFlow-based neural network library

HIPS / autograd

Efficiently computes derivatives of NumPy code.

open-edge-platform / anomalib

An anomaly detection library comprising state-of-the-art algorithms and features such as experiment management, hyper-parameter optimization, and edge inference.

awslabs / gluonts

Probabilistic time series modeling in Python

NVIDIAGameWorks / kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

pytorch / executorch

On-device AI across mobile, embedded and edge for PyTorch

sktime / pytorch-forecasting

Time series forecasting with PyTorch

pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

google-deepmind / dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

ourownstory / neural_prophet

NeuralProphet: A simple forecasting package

Nixtla / neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

google / deepvariant

DeepVariant is an analysis pipeline that uses a deep neural network to call genetic variants from next-generation DNA sequencing data.

truera / trulens

Evaluation and Tracking for LLM Experiments and AI Agents

PennyLaneAI / pennylane

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.

mljar / mljar-supervised

Python package for AutoML on Tabular Data with Feature Engineering, Hyper-Parameters Tuning, Explanations and Automatic Documentation

google-deepmind / dm-haiku

JAX-based neural network library

qualcomm / aimet

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.

chemprop / chemprop

Message Passing Neural Networks for Molecule Property Prediction

ailia-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK

AIStream-Peelout / flow-forecast

Deep learning PyTorch library for time series forecasting, classification, and anomaly detection (originally for flood forecasting).

ahmedfgad / GeneticAlgorithmPython

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

fastmachinelearning / hls4ml

Machine learning on FPGAs using HLS

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

WenjieDu / PyPOTS

A Python toolkit/library for reality-centric machine/deep learning & data mining on partially-observed time series, with 50+ SOTA neural network models for scientific analysis tasks (imputation, classification, clustering, forecasting, anomaly detection, cleaning) on incomplete industrial irregularly-sampled multivariate TS with NaN missing values

jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python

youngfish42 / Awesome-FL

Comprehensive and timely academic information on federated learning (papers, frameworks, datasets, tutorials, workshops)

meta-pytorch / opacus

Training PyTorch models with differential privacy

kyegomez / BitNet

Implementation of "BitNet: Scaling 1-bit Transformers for Large Language Models" in pytorch

c2g-dev / city2graph

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

meta-pytorch / tnt

A lightweight library for PyTorch training tools and utilities

BindsNET / bindsnet

Simulation of spiking neural networks (SNNs) using PyTorch.

InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

JustGlowing / minisom

:red_circle: MiniSom is a minimalistic implementation of the Self Organizing Maps

google / uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

Xilinx / brevitas

Brevitas: neural network quantization in PyTorch

tensorflow / model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

bgavran / Category_Theory_Machine_Learning

List of papers studying machine learning through the lens of category theory

geomstats / geomstats

Computations and statistics on manifolds with geometric structures.

sepandhaghighi / pycm

Multi-class confusion matrix library in Python

aeon-toolkit / aeon

A toolkit for time series machine learning and deep learning

open-edge-platform / geti

Build, train, optimize, and run computer vision models locally, from raw images to live inference. Open source, optimized for Intel XPU (CPU-only and CUDA also supported).