Best Machine Learning & Data Science AI GitHub Repos

Ollama

AI GitHub Repos

#10

Open-source tool for running large language models on your own computer. It lets developers install models like Llama, Gemma, Qwen and DeepSeek with a single command,…Ollama is an open-source tool for running large language models on your own computer. It lets developers install models like Llama, Gemma, Qwen and DeepSeek with a single command,…

LlamaFactory

AI GitHub Repos

#12

GitHub repository focused on the unified and efficient fine-tuning of over 100 large language models (LLMs) and vision-language models (VLMs). It aims to…LlamaFactory is a GitHub repository focused on the unified and efficient fine-tuning of over 100 large language models (LLMs) and vision-language models (VLMs). It aims to…

PaddleOCR

AI GitHub Repos

#14

Powerful, lightweight OCR toolkit designed to convert PDFs and image documents into structured data for AI applications. It supports over 100 languages and bridges…PaddleOCR is a powerful, lightweight OCR toolkit designed to convert PDFs and image documents into structured data for AI applications. It supports over 100 languages and bridges…

Ultralytics YOLO

AI GitHub Repos

#18

Includes YOLO26, YOLO11, and YOLOv8, providing capabilities for object detection, instance segmentation, semantic segmentation, image classification, pose…Ultralytics YOLO includes YOLO26, YOLO11, and YOLOv8, providing capabilities for object detection, instance segmentation, semantic segmentation, image classification, pose…

Langfuse

AI GitHub Repos

#19

Open-source AI engineering platform offering LLM evaluations, observability, metrics, prompt management, and more. It integrates with OpenTelemetry, LangChain, and…Langfuse is an open-source AI engineering platform offering LLM evaluations, observability, metrics, prompt management, and more. It integrates with OpenTelemetry, LangChain, and…

prompts.chat

AI GitHub Repos

#21

Open-source library of prompts for AI chat models, compatible with ChatGPT, Claude, Gemini, Llama, and Mistral. It's free to…prompts.chat (formerly Awesome ChatGPT Prompts) is an open-source library of prompts for AI chat models, compatible with ChatGPT, Claude, Gemini, Llama, and Mistral. It's free to…

vLLM

AI GitHub Repos

#23

High-throughput and memory-efficient inference and serving engine designed for large language models (LLMs). It optimizes performance while minimizing…vllm-project/vllm is a high-throughput and memory-efficient inference and serving engine designed for large language models (LLMs). It optimizes performance while minimizing…

Chroma

AI GitHub Repos

#24

AI-focused search infrastructure project hosted on GitHub. It enables developers to contribute to its development and enhance AI search capabilities through…Chroma is an AI-focused search infrastructure project hosted on GitHub. It enables developers to contribute to its development and enhance AI search capabilities through…

RAGFlow

AI GitHub Repos

#29

Open-source Retrieval-Augmented Generation (RAG) engine that integrates advanced RAG with Agent capabilities, providing a robust context layer for large language…RAGFlow is an open-source Retrieval-Augmented Generation (RAG) engine that integrates advanced RAG with Agent capabilities, providing a robust context layer for large language…

Unsloth

AI GitHub Repos

#33

Local user interface designed to run and train large language models (LLMs) and diffusion models. It supports various models including Qwen3.8, Kimi K3,…unslothai/unsloth is a local user interface designed to run and train large language models (LLMs) and diffusion models. It supports various models including Qwen3.8, Kimi K3,…

Open-source tool for vector similarity search in Postgres. It allows developers to efficiently manage and query vector data, enhancing applications that rely on…pgvector is an open-source tool for vector similarity search in Postgres. It allows developers to efficiently manage and query vector data, enhancing applications that rely on…

Milvus - High-Performance Vector Database

AI GitHub Repos

#39

High-performance, cloud-native vector database designed for scalable vector approximate nearest neighbor (ANN) search, enabling efficient data retrieval and management…Milvus is a high-performance, cloud-native vector database designed for scalable vector approximate nearest neighbor (ANN) search, enabling efficient data retrieval and management…

Fabric AI Framework

AI GitHub Repos

#42

Open-source framework designed to augment human capabilities using AI. It offers a modular system to address specific problems through a crowdsourced set of AI…Fabric is an open-source framework designed to augment human capabilities using AI. It offers a modular system to address specific problems through a crowdsourced set of AI…

LocalAI

AI GitHub Repos

#43

Open-source AI engine that allows users to run various models, including LLMs, vision, voice, image, and video, on any hardware without requiring a GPU. This…LocalAI is an open-source AI engine that allows users to run various models, including LLMs, vision, voice, image, and video, on any hardware without requiring a GPU. This…

SGLang

AI GitHub Repos

#53

High-performance serving framework designed for large language models and multimodal models. It provides efficient serving capabilities that enhance the performance of…SGLang is a high-performance serving framework designed for large language models and multimodal models. It provides efficient serving capabilities that enhance the performance of…

text-generation-webui

AI GitHub Repos

#57

Open-source desktop application designed for local large language models (LLMs). It supports text, vision, and tool-calling functionalities, and is…text-generation-webui is an open-source desktop application designed for local large language models (LLMs). It supports text, vision, and tool-calling functionalities, and is…

ColossalAI

AI GitHub Repos

#60

Project aimed at making large AI models more affordable, faster, and accessible. It provides tools and frameworks to optimize AI model training and deployment,…ColossalAI is a project aimed at making large AI models more affordable, faster, and accessible. It provides tools and frameworks to optimize AI model training and deployment,…

TensorRT-LLM

AI GitHub Repos

#64

Provides a Python API for defining Large Language Models, optimizing inference on NVIDIA GPUs. It includes components for creating Python and C++ runtimes to…TensorRT-LLM provides a Python API for defining Large Language Models, optimizing inference on NVIDIA GPUs. It includes components for creating Python and C++ runtimes to…

Netron Visualizer

AI GitHub Repos

#69

Visualizer for neural network, deep learning, and machine learning models. It provides an intuitive interface to explore model architectures and parameters, making it…Netron is a visualizer for neural network, deep learning, and machine learning models. It provides an intuitive interface to explore model architectures and parameters, making it…

LMDeploy

AI GitHub Repos

#73

Toolkit designed for compressing, deploying, and serving large language models (LLMs). It provides developers with the necessary tools to efficiently manage LLMs,…LMDeploy is a toolkit designed for compressing, deploying, and serving large language models (LLMs). It provides developers with the necessary tools to efficiently manage LLMs,…

Flair NLP Framework

AI GitHub Repos

#75

Simple framework for state-of-the-art Natural Language Processing (NLP) tasks. It offers easy-to-use interfaces for text classification, sequence labeling, and more,…Flair is a simple framework for state-of-the-art Natural Language Processing (NLP) tasks. It offers easy-to-use interfaces for text classification, sequence labeling, and more,…

PaddleX

AI GitHub Repos

#77

All-in-one development tool based on PaddlePaddle, designed to streamline AI model development and deployment. It offers a comprehensive suite of features for…PaddleX is an all-in-one development tool based on PaddlePaddle, designed to streamline AI model development and deployment. It offers a comprehensive suite of features for…

DeepEval

AI GitHub Repos

#80

Open-source framework designed for evaluating large language models (LLMs). It allows developers to contribute and improve the evaluation process, enhancing the…DeepEval is an open-source framework designed for evaluating large language models (LLMs). It allows developers to contribute and improve the evaluation process, enhancing the…

llmware

AI GitHub Repos

#81

Unified framework designed for building enterprise RAG pipelines using small, specialized models. It simplifies the integration of various AI models into cohesive…llmware is a unified framework designed for building enterprise RAG pipelines using small, specialized models. It simplifies the integration of various AI models into cohesive…

Stability Matrix

AI GitHub Repos

#83

Multi-platform package manager and inference UI for Stable Diffusion. It offers one-click installation and updates for various Stable Diffusion Web UI…Stability Matrix is a multi-platform package manager and inference UI for Stable Diffusion. It offers one-click installation and updates for various Stable Diffusion Web UI…

Phoenix

AI GitHub Repos

#85

Arize Phoenix is an AI observability and evaluation tool designed to help developers monitor and assess AI models. It provides insights into model performance, enabling better…

Feast Feature Store

AI GitHub Repos

#90

Open-source feature store designed for AI and machine learning applications. It enables data scientists to manage and serve features to models in production, enhancing…Feast is an open-source feature store designed for AI and machine learning applications. It enables data scientists to manage and serve features to models in production, enhancing…

BentoML

AI GitHub Repos

#93

Open-source platform designed to simplify the deployment of AI models and applications. It enables developers to build model inference APIs, manage job queues, and…BentoML is an open-source platform designed to simplify the deployment of AI models and applications. It enables developers to build model inference APIs, manage job queues, and…

Mindshub

AI GitHub Repos

#99

Unified workspace on GitHub where open-source models can be efficiently managed and utilized. It provides a platform for developers to collaborate and integrate…Mindshub is a unified workspace on GitHub where open-source models can be efficiently managed and utilized. It provides a platform for developers to collaborate and integrate…

Axolotl

AI GitHub Repos

#101

Axolotl AI Cloud is a GitHub repository that provides a platform for AI model development and deployment. It allows users to contribute to AI projects, enhancing collaboration and…

NannyML

AI GitHub Repos

#106

Python library focused on post-deployment data science. It helps data scientists monitor and analyze model performance after deployment, ensuring reliability and…NannyML is a Python library focused on post-deployment data science. It helps data scientists monitor and analyze model performance after deployment, ensuring reliability and…

Detectron2 Platform

AI GitHub Repos

#113

Platform designed for object detection, segmentation, and various visual recognition tasks. Developed by Facebook Research, it provides advanced tools for…Detectron2 is a platform designed for object detection, segmentation, and various visual recognition tasks. Developed by Facebook Research, it provides advanced tools for…

Open-source text annotation tool designed for machine learning practitioners. It supports various labeling tasks and data formats, offering REST APIs for integration…doccano is an open-source text annotation tool designed for machine learning practitioners. It supports various labeling tasks and data formats, offering REST APIs for integration…

Cognee

AI GitHub Repos

#123

Open-source AI memory platform designed for agents, enabling persistent long-term memory across sessions with a self-hosted knowledge graph engine. Ideal for…Cognee is an open-source AI memory platform designed for agents, enabling persistent long-term memory across sessions with a self-hosted knowledge graph engine. Ideal for…

Xtreme1

AI GitHub Repos

#124

Open-source platform designed for data labeling and annotation, supporting multimodal AI training data. It handles 3D LiDAR point cloud, image, and LLM data, making…Xtreme1 is an open-source platform designed for data labeling and annotation, supporting multimodal AI training data. It handles 3D LiDAR point cloud, image, and LLM data, making…

Voxel51 FiftyOne

AI GitHub Repos

#128

FiftyOne is an open-source tool for refining high-quality datasets and visual AI models. It allows developers to manage, visualize, and analyze datasets efficiently, enhancing AI…

DeepFace

AI GitHub Repos

#135

Lightweight library for Python that offers face recognition and facial attribute analysis, including age, gender, emotion, and race detection. It is designed to be…DeepFace is a lightweight library for Python that offers face recognition and facial attribute analysis, including age, gender, emotion, and race detection. It is designed to be…

llamafile

AI GitHub Repos

#136

Tool designed to simplify the distribution and execution of large language models (LLMs) using a single file. It facilitates collaboration and development within…Llamafile is a tool designed to simplify the distribution and execution of large language models (LLMs) using a single file. It facilitates collaboration and development within…

Dalai

AI GitHub Repos

#141

GitHub project that simplifies running the LLaMA model on local machines. It provides an accessible way for developers to experiment with LLaMA without needing…Dalai is a GitHub project that simplifies running the LLaMA model on local machines. It provides an accessible way for developers to experiment with LLaMA without needing…

YOLO-World

AI GitHub Repos

#142

Real-time open-vocabulary object detection tool developed by AILab-CVC. It aims to enhance object detection capabilities by allowing detection of objects not seen…YOLO-World is a real-time open-vocabulary object detection tool developed by AILab-CVC. It aims to enhance object detection capabilities by allowing detection of objects not seen…

SGLang-Omni

AI GitHub Repos

#155This week

High-performance serving framework designed for audio models like TTS and ASR, as well as unified multimodal models. It provides robust support for deploying and…SGLang-Omni is a high-performance serving framework designed for audio models like TTS and ASR, as well as unified multimodal models. It provides robust support for deploying and…

easy-dataset

AI GitHub Repos

#157

Tool designed to streamline the creation of datasets for LLM fine-tuning, RAG, and evaluation. It simplifies the process of preparing data, making it accessible…Easy-dataset is a tool designed to streamline the creation of datasets for LLM fine-tuning, RAG, and evaluation. It simplifies the process of preparing data, making it accessible…

MLX LM

AI GitHub Repos

#162

GitHub project designed to facilitate the running of large language models (LLMs) using the MLX framework. It allows developers to contribute to its development and…MLX LM is a GitHub project designed to facilitate the running of large language models (LLMs) using the MLX framework. It allows developers to contribute to its development and…

Caveman

AI GitHub Repos

#180

GitHub project that optimizes token usage for coding agents by mimicking caveman speech. It reduces token consumption by 65%, enhancing efficiency for developers…Caveman is a GitHub project that optimizes token usage for coding agents by mimicking caveman speech. It reduces token consumption by 65%, enhancing efficiency for developers…

FailproofAI

AI GitHub Repos

#181

Provides observability and enforcement for AI agent harnesses, ensuring every run is captured and runtime reliability is maintained with policy enforcement. It is…FailproofAI provides observability and enforcement for AI agent harnesses, ensuring every run is captured and runtime reliability is maintained with policy enforcement. It is…

TokenSpeed

AI GitHub Repos

#188This week

High-performance LLM inference engine designed to deliver rapid processing speeds. It is optimized for large language models, providing efficient and scalable…TokenSpeed is a high-performance LLM inference engine designed to deliver rapid processing speeds. It is optimized for large language models, providing efficient and scalable…

RagaAI Catalyst

AI GitHub Repos

#200

Python SDK designed for AI observability, monitoring, and evaluation. It offers features like agent and tool tracing, debugging multi-agent systems, and…RagaAI Catalyst is a Python SDK designed for AI observability, monitoring, and evaluation. It offers features like agent and tool tracing, debugging multi-agent systems, and…

PowerInfer

AI GitHub Repos

#201

High-speed large language model serving tool designed for local deployment. It enables efficient and rapid inference of large language models, making it suitable…PowerInfer is a high-speed large language model serving tool designed for local deployment. It enables efficient and rapid inference of large language models, making it suitable…

Ollaya

AI GitHub Repos

#213

GitHub project that enables users to run open decision models locally. It supports Laya, decider, NLI, and GLiClass models, providing a TypeSafe-compatible API for…Ollaya is a GitHub project that enables users to run open decision models locally. It supports Laya, decider, NLI, and GLiClass models, providing a TypeSafe-compatible API for…

labelCloud

AI GitHub Repos

#219

Lightweight tool designed for labeling 3D bounding boxes in point clouds. It is ideal for developers and researchers working with 3D data, providing an efficient…labelCloud is a lightweight tool designed for labeling 3D bounding boxes in point clouds. It is ideal for developers and researchers working with 3D data, providing an efficient…

AirLLM

AI GitHub Repos

#236

GitHub project that enables 70 billion parameter inference using a single 4GB GPU. It is designed to optimize large language model performance on limited hardware,…AirLLM is a GitHub project that enables 70 billion parameter inference using a single 4GB GPU. It is designed to optimize large language model performance on limited hardware,…

Rocketride Server

AI GitHub Repos

#237

High-performance AI pipeline engine with a C++ core. It supports over 50 Python-extensible nodes, enabling the building, debugging, and scaling of LLM…Rocketride Server is a high-performance AI pipeline engine with a C++ core. It supports over 50 Python-extensible nodes, enabling the building, debugging, and scaling of LLM…

R2R — AI Retrieval System

AI GitHub Repos

#240

State-of-the-art AI retrieval system designed for efficient data retrieval and generation. It utilizes Agentic Retrieval-Augmented Generation (RAG) with a RESTful API,…R2R is a state-of-the-art AI retrieval system designed for efficient data retrieval and generation. It utilizes Agentic Retrieval-Augmented Generation (RAG) with a RESTful API,…

Project Maya

AI GitHub Repos

#250This week

Allows users to run the GLM-5.3-Flash model privately on their own GPUs, supporting both NVIDIA and AMD hardware. It offers an engine, server, dashboard, and APIs…Project Maya allows users to run the GLM-5.3-Flash model privately on their own GPUs, supporting both NVIDIA and AMD hardware. It offers an engine, server, dashboard, and APIs…

GPUStack

AI GitHub Repos

#116

GPU cluster manager designed for high-performance AI model serving, including vLLM and SGLang, and provides on-demand SSH-accessible GPU instances. It facilitates…GPUStack is a GPU cluster manager designed for high-performance AI model serving, including vLLM and SGLang, and provides on-demand SSH-accessible GPU instances. It facilitates…

Strata

AI GitHub Repos

#23

Inference engine that enables the deployment of Qwen3.8-Flash-Next on consumer hardware. It supports one-click installation on Windows and Linux, and offers…Strata is an inference engine that enables the deployment of Qwen3.8-Flash-Next on consumer hardware. It supports one-click installation on Windows and Linux, and offers…

SD-Trainer

AI GitHub Repos

#96

Provides LoRA and Dreambooth training scripts with a GUI, utilizing kohya-ss's trainer for diffusion models. It is designed for developers and researchers working with…SD-Trainer provides LoRA and Dreambooth training scripts with a GUI, utilizing kohya-ss's trainer for diffusion models. It is designed for developers and researchers working with…

Cactus Compute

AI GitHub Repos

#21

Open-source project focused on quantization, kernels, runtime, and inference engines for mobile devices, wearables, smart home systems, and robots. It aims to…Cactus Compute is an open-source project focused on quantization, kernels, runtime, and inference engines for mobile devices, wearables, smart home systems, and robots. It aims to…

Laya-MLX

AI GitHub Repos

#136

Laya MLX is a native runtime designed for Laya typed decision models, offering rapid decision-making capabilities on M3 Max. It operates without text generation, PyTorch, or cloud…

Splash

AI GitHub Repos

#137

Local inference engine designed specifically for Apple silicon. It enables efficient model inference on Apple devices, providing a streamlined solution for developers…Splash is a local inference engine designed specifically for Apple silicon. It enables efficient model inference on Apple devices, providing a streamlined solution for developers…

Suna

AI GitHub Repos

#120

Open-source AI operating system designed to facilitate AI development and deployment. It offers a collaborative platform for developers to contribute and enhance AI…Suna is an open-source AI operating system designed to facilitate AI development and deployment. It offers a collaborative platform for developers to contribute and enhance AI…

Oumi

AI GitHub Repos

#145

GitHub project designed to fine-tune, evaluate, and deploy large language models like Qwen and Gemma. It supports open agentic LLMs and VLMs, offering tools for both…Oumi is a GitHub project designed to fine-tune, evaluate, and deploy large language models like Qwen and Gemma. It supports open agentic LLMs and VLMs, offering tools for both…

Sherpa-ONNX

AI GitHub Repos

#71

Open-source project offering speech-to-text, text-to-speech, speaker diarization, and more using Kaldi with onnxruntime. It supports multiple platforms and…Sherpa-ONNX is an open-source project offering speech-to-text, text-to-speech, speaker diarization, and more using Kaldi with onnxruntime. It supports multiple platforms and…

Kangas Multimedia Dataset Explorer

AI GitHub Repos

Tool designed to explore multimedia datasets at scale. It allows users to efficiently manage and analyze large datasets, facilitating data-driven insights and…Kangas is a tool designed to explore multimedia datasets at scale. It allows users to efficiently manage and analyze large datasets, facilitating data-driven insights and…

omlx

AI GitHub Repos

#138

LLM inference server designed for Apple Silicon, featuring continuous batching and SSD caching. It is managed directly from the macOS menu bar, providing an efficient…omlx is an LLM inference server designed for Apple Silicon, featuring continuous batching and SSD caching. It is managed directly from the macOS menu bar, providing an efficient…

DataGym.ai

AI GitHub Repos

DataGym is an open-source tool for annotating and labeling image and video assets. It is designed to facilitate efficient data preparation for machine learning projects, offering…

TensorFlow

AI GitHub Repos

Open-source machine learning framework designed for everyone. It provides a comprehensive ecosystem for building and deploying machine learning models, making it…TensorFlow is an open-source machine learning framework designed for everyone. It provides a comprehensive ecosystem for building and deploying machine learning models, making it…

Hugging Face Transformers

AI GitHub Repos

🤗 Transformers is a model-definition framework designed for state-of-the-art machine learning models across text, vision, audio, and multimodal domains, facilitating both…

llama.cpp

AI GitHub Repos

GitHub repository focused on LLM inference in C/C++. It allows developers to contribute to its development and utilize its capabilities for machine…ggml-org/llama.cpp is a GitHub repository focused on LLM inference in C/C++. It allows developers to contribute to its development and utilize its capabilities for machine…

PyTorch

AI GitHub Repos

Open-source machine learning library that provides tools for building dynamic neural networks in Python, leveraging strong GPU acceleration for efficient computation.PyTorch is an open-source machine learning library that provides tools for building dynamic neural networks in Python, leveraging strong GPU acceleration for efficient computation.

Keras

AI GitHub Repos

Open-source deep learning library designed for humans. It simplifies the process of building neural networks and allows developers to contribute to its development on…Keras is an open-source deep learning library designed for humans. It simplifies the process of building neural networks and allows developers to contribute to its development on…

MLX

AI GitHub Repos

Array framework designed specifically for Apple silicon, enabling efficient machine learning operations. It allows developers to contribute to its development on GitHub,…MLX is an array framework designed specifically for Apple silicon, enabling efficient machine learning operations. It allows developers to contribute to its development on GitHub,…

nanoGPT

AI GitHub Repos

Minimalist, high-performance GitHub repository for training and fine-tuning medium-sized GPT models. It offers a simple, readable codebase for researchers and…nanoGPT is a minimalist, high-performance GitHub repository for training and fine-tuning medium-sized GPT models. It offers a simple, readable codebase for researchers and…

LlamaIndex

AI GitHub Repos

Open source Python framework for connecting large language models to a user's own data. It provides connectors, indices, and a retrieval interface for…LlamaIndex is an open source Python framework for connecting large language models to a user's own data. It provides connectors, indices, and a retrieval interface for…

Unlimited OCR Works

AI GitHub Repos

GitHub repository by Baidu focused on optical character recognition. It introduces one-shot long-horizon parsing, aiming to enhance OCR capabilities…Unlimited OCR is a GitHub repository by Baidu focused on optical character recognition. It introduces one-shot long-horizon parsing, aiming to enhance OCR capabilities…

Shumai

AI GitHub Repos

Fast, differentiable tensor library for JavaScript and TypeScript, built with Bun and Flashlight. It enables software engineers and researchers to create datasets,…Shumai is a fast, differentiable tensor library for JavaScript and TypeScript, built with Bun and Flashlight. It enables software engineers and researchers to create datasets,…

Seldon Core

AI GitHub Repos

MLOps and LLMOps framework for deploying, managing, and scaling AI systems on Kubernetes. It enables standardized deployment of various model types across…Seldon Core is an MLOps and LLMOps framework for deploying, managing, and scaling AI systems on Kubernetes. It enables standardized deployment of various model types across…

FaceNet

AI GitHub Repos

TensorFlow implementation for face recognition and clustering, based on the FaceNet paper. It leverages deep learning models to generate unified embeddings for faces,…FaceNet is a TensorFlow implementation for face recognition and clustering, based on the FaceNet paper. It leverages deep learning models to generate unified embeddings for faces,…

DGCNN

AI GitHub Repos

AI model for learning on point clouds, implementing Dynamic Graph CNN. It achieves state-of-the-art performance in tasks like classification and segmentation of 3D…DGCNN is an AI model for learning on point clouds, implementing Dynamic Graph CNN. It achieves state-of-the-art performance in tasks like classification and segmentation of 3D…

Mamba SSM Architecture

AI GitHub Repos

Novel state space model architecture designed for efficient sequence modeling, particularly effective on information-dense data like language. It offers a…Mamba is a novel state space model architecture designed for efficient sequence modeling, particularly effective on information-dense data like language. It offers a…

N-BEATS

AI GitHub Repos

Neural-network based model for univariate time-series forecasting. Developed by ServiceNow Research, it implements the N-BEATS algorithm for reproducible experimental…N-BEATS is a neural-network based model for univariate time-series forecasting. Developed by ServiceNow Research, it implements the N-BEATS algorithm for reproducible experimental…

ViT-Adapter

AI GitHub Repos

AI model that enhances Vision Transformer (ViT) performance for dense prediction tasks like object detection and segmentation. It introduces image-specific…ViT-Adapter is an AI model that enhances Vision Transformer (ViT) performance for dense prediction tasks like object detection and segmentation. It introduces image-specific…

DM-GAN

AI GitHub Repos

PyTorch implementation of Dynamic Memory Generative Adversarial Networks for text-to-image synthesis. This repository provides code, pretrained models, and evaluation…DM-GAN is a PyTorch implementation of Dynamic Memory Generative Adversarial Networks for text-to-image synthesis. This repository provides code, pretrained models, and evaluation…

OpenLLaMA

AI GitHub Repos

Permissively licensed, open-source reproduction of Meta AI's LLaMA 7B model. Trained on the RedPajama dataset, it offers 3B, 7B, and 13B parameter versions,…OpenLLaMA is a permissively licensed, open-source reproduction of Meta AI's LLaMA 7B model. Trained on the RedPajama dataset, it offers 3B, 7B, and 13B parameter versions,…

Generative Image-to-Text Transformer

AI GitHub Repos

GIT (Generative Image-to-text Transformer) is an AI model by Microsoft for vision and language tasks. It generates text descriptions from images and can perform visual question…

UNITER

AI GitHub Repos

Research code repository for the ECCV 2020 paper 'UNITER: UNiversal Image-TExt Representation Learning'. It provides code for finetuning and inference across various…UNITER is a research code repository for the ECCV 2020 paper 'UNITER: UNiversal Image-TExt Representation Learning'. It provides code for finetuning and inference across various…

Principal Neighbourhood Aggregation

AI GitHub Repos

This repository provides an implementation of Principal Neighbourhood Aggregation (PNA) for Graph Neural Networks. It supports PyTorch, DGL, and PyTorch Geometric frameworks,…

DeepGCNs

AI GitHub Repos

This GitHub repository provides PyTorch implementations for DeepGCNs, DeeperGCN, and GNN1000. It enables training very deep Graph Convolutional Networks by adapting concepts from…

Generative adversarial network for efficient and high-fidelity speech synthesis. It models periodic patterns in audio to enhance sample quality, achieving human-like…HiFi-GAN is a generative adversarial network for efficient and high-fidelity speech synthesis. It models periodic patterns in audio to enhance sample quality, achieving human-like…

Open-source toolkit for end-to-end speech processing. It provides comprehensive recipes and tools for tasks like Automatic Speech Recognition (ASR), Text-to-Speech…ESPnet is an open-source toolkit for end-to-end speech processing. It provides comprehensive recipes and tools for tasks like Automatic Speech Recognition (ASR), Text-to-Speech…

Vision Transformer

AI GitHub Repos

The Vision Transformer (ViT) repository provides models and code for image recognition tasks. It includes implementations of the Vision Transformer and MLP-Mixer architectures,…

FNet

AI GitHub Repos

Efficient Transformer-like encoder architecture that replaces self-attention with Fourier Transforms. Developed by Google Research, it offers a high-performance…FNet is an efficient Transformer-like encoder architecture that replaces self-attention with Fourier Transforms. Developed by Google Research, it offers a high-performance…

SpanBERT

AI GitHub Repos

AI model focused on improving pre-training by representing and predicting spans. It offers pre-trained base and large cased models, compatible with HuggingFace BERT…SpanBERT is an AI model focused on improving pre-training by representing and predicting spans. It offers pre-trained base and large cased models, compatible with HuggingFace BERT…

LayoutLM

AI GitHub Repos

Multimodal pre-training model for visually-rich document understanding and information extraction. It combines text, layout, and image information to achieve…LayoutLM is a multimodal pre-training model for visually-rich document understanding and information extraction. It combines text, layout, and image information to achieve…

MASS

AI GitHub Repos

Pre-training method for sequence-to-sequence language generation tasks. It masks sentence fragments for the encoder to predict in the decoder, enhancing tasks like…MASS is a pre-training method for sequence-to-sequence language generation tasks. It masks sentence fragments for the encoder to predict in the decoder, enhancing tasks like…

ProphetNet

AI GitHub Repos

Research project by the MSRA NLC team focused on natural language generation. It provides official implementations of pretrained models, including those for future…ProphetNet is a research project by the MSRA NLC team focused on natural language generation. It provides official implementations of pretrained models, including those for future…

Funnel-Transformer

AI GitHub Repos

AI model that compresses hidden states to reduce computation cost. It allows for deeper or wider models with the same FLOPs and can recover token-level…Funnel-Transformer is an AI model that compresses hidden states to reduce computation cost. It allows for deeper or wider models with the same FLOPs and can recover token-level…

ConvBERT

AI GitHub Repos

Open-source AI model for pre-training language models, introducing a novel architecture with span-based dynamic convolution. This GitHub repository provides the…ConvBERT is an open-source AI model for pre-training language models, introducing a novel architecture with span-based dynamic convolution. This GitHub repository provides the…

MPNet

AI GitHub Repos

Novel pre-training method for language understanding tasks, improving upon BERT and XLNet. It offers a unified implementation for various pre-training models and…MPNet is a novel pre-training method for language understanding tasks, improving upon BERT and XLNet. It offers a unified implementation for various pre-training models and…

How Attentive are Graph Attention Networks?

AI GitHub Repos

This repository provides the official code implementation for the research paper "How Attentive are Graph Attention Networks?" published at ICLR'2022. It offers reproducible…

SEAL Link Prediction

AI GitHub Repos

Novel framework for link prediction that transforms the problem into a subgraph classification task. It extracts h-hop enclosing subgraphs and node information matrices,…SEAL is a novel framework for link prediction that transforms the problem into a subgraph classification task. It extracts h-hop enclosing subgraphs and node information matrices,…

GraphSAINT

AI GitHub Repos

Flexible framework for training deep Graph Neural Networks (GNNs) on large graphs. It introduces a novel minibatch training method using graph sampling, offering…GraphSAINT is a flexible framework for training deep Graph Neural Networks (GNNs) on large graphs. It introduces a novel minibatch training method using graph sampling, offering…

StyleSwin: Transformer-based GAN

AI GitHub Repos

Transformer-based Generative Adversarial Network (GAN) designed for high-resolution image generation. It leverages Swin transformers and a novel double attention…StyleSwin is a transformer-based Generative Adversarial Network (GAN) designed for high-resolution image generation. It leverages Swin transformers and a novel double attention…

Voicebox

AI GitHub Repos

Open-source AI voice studio that allows users to clone, dictate, and create voice projects. It provides a platform for voice synthesis and manipulation, making it…Voicebox is an open-source AI voice studio that allows users to clone, dictate, and create voice projects. It provides a platform for voice synthesis and manipulation, making it…

TabNet

AI GitHub Repos

PyTorch implementation of the TabNet paper, offering an attentive and interpretable approach to tabular data learning. It supports classification, regression, and multi-task…

Neural Oblivious Decision Ensembles

AI GitHub Repos

Python library for deep learning on tabular data. It implements ensembles of oblivious, differentiable decision trees, offering a…Neural Oblivious Decision Ensembles (NODE) is a Python library for deep learning on tabular data. It implements ensembles of oblivious, differentiable decision trees, offering a…

Deep Learning Models for Tabular Data

AI GitHub Repos

This GitHub repository provides the official implementation for the NeurIPS 2021 paper 'Revisiting Deep Learning Models for Tabular Data.' It explores deep learning architectures…

SAINT

AI GitHub Repos

This repository provides the official PyTorch implementation of the SAINT model, designed for improved neural networks on tabular data. It leverages row attention and contrastive…

R-FCN

AI GitHub Repos

Region-based object detection framework that utilizes fully convolutional networks for accurate and efficient image analysis. It shares computation across the entire…R-FCN is a region-based object detection framework that utilizes fully convolutional networks for accurate and efficient image analysis. It shares computation across the entire…

SPP_net

AI GitHub Repos

Re-implementation of the Spatial Pyramid Pooling algorithm for deep convolutional networks in visual recognition. It aims to reproduce object detection results from a…SPP_net is a re-implementation of the Spatial Pyramid Pooling algorithm for deep convolutional networks in visual recognition. It aims to reproduce object detection results from a…

Squeeze-and-Excitation Networks

AI GitHub Repos

Deep learning model architecture that adaptively recalibrates channel-wise feature responses. It enhances performance by explicitly…Squeeze-and-Excitation Networks (SENet) is a deep learning model architecture that adaptively recalibrates channel-wise feature responses. It enhances performance by explicitly…

ShuffleNet Series

AI GitHub Repos

The ShuffleNet Series repository offers a collection of highly efficient convolutional neural network models designed for mobile devices and resource-constrained environments. It…

ConvMixer

AI GitHub Repos

This repository provides an implementation of ConvMixer, a convolutional neural network architecture for image recognition tasks. It is based on the paper "Patches Are All You…

RepVGG

AI GitHub Repos

Powerful and simple ConvNet architecture that achieves high accuracy on ImageNet. It utilizes a VGG-style design with re-parameterization techniques, allowing for…RepVGG is a powerful and simple ConvNet architecture that achieves high accuracy on ImageNet. It utilizes a VGG-style design with re-parameterization techniques, allowing for…

Vision GNN (ViG)

AI GitHub Repos

PyTorch implementation of Vision Graph Neural Networks developed by Huawei Noah's Ark Lab. It offers efficient AI backbones for image processing tasks,…Vision GNN (ViG) is a PyTorch implementation of Vision Graph Neural Networks developed by Huawei Noah's Ark Lab. It offers efficient AI backbones for image processing tasks,…

Deformable ConvNets

AI GitHub Repos

Deformable Convolutional Networks (DCN) enhance deep learning models by adaptively learning sampling offsets for convolutional kernels. This allows networks to better capture…

Skip-Thought Vectors

AI GitHub Repos

Provides Sent2Vec encoder and training code based on the research paper "Skip-Thought Vectors." It enables the generation of dense vector representations for…Skip-Thought Vectors provides Sent2Vec encoder and training code based on the research paper "Skip-Thought Vectors." It enables the generation of dense vector representations for…

SNGAN Projection

AI GitHub Repos

This GitHub repository provides an official Chainer implementation for conditional image generation. It utilizes spectral normalization and a projection discriminator for…

PyTorch MADE

AI GitHub Repos

This repository provides a PyTorch implementation of MADE (Masked Autoencoder Density Estimation). It enables turning auto-encoders into autoregressive density models by masking…

StyleGAN-XL

AI GitHub Repos

AI model for generating high-resolution images from large, diverse datasets. It scales StyleGAN architecture for improved image synthesis quality and diversity.…StyleGAN-XL is an AI model for generating high-resolution images from large, diverse datasets. It scales StyleGAN architecture for improved image synthesis quality and diversity.…

DimeNet and DimeNet++

AI GitHub Repos

Advanced neural network models for molecular graphs, focusing on directional message passing. These models, proposed in ICLR 2020 and NeurIPS-W 2020,…DimeNet and DimeNet++ are advanced neural network models for molecular graphs, focusing on directional message passing. These models, proposed in ICLR 2020 and NeurIPS-W 2020,…

LSTNet

AI GitHub Repos

Deep learning model designed for time series forecasting. It effectively models both long-term and short-term temporal patterns using a combination of recurrent neural…LSTNet is a deep learning model designed for time series forecasting. It effectively models both long-term and short-term temporal patterns using a combination of recurrent neural…

Autoformer

AI GitHub Repos

AI model for long-term time series forecasting, leveraging decomposition transformers and auto-correlation mechanisms. It achieves state-of-the-art results across…Autoformer is an AI model for long-term time series forecasting, leveraging decomposition transformers and auto-correlation mechanisms. It achieves state-of-the-art results across…

Informer2020

AI GitHub Repos

The Informer2020 GitHub repository provides the PyTorch implementation for the Informer model, designed for efficient long sequence time-series forecasting. It features ProbSparse…

FEDformer

AI GitHub Repos

Frequency Enhanced Decomposed Transformer designed for efficient long-term time series forecasting. It achieves linear complexity with sequence length,…FEDformer is a Frequency Enhanced Decomposed Transformer designed for efficient long-term time series forecasting. It achieves linear complexity with sequence length,…

LTSF-Linear

AI GitHub Repos

Official PyTorch implementation for "Are Transformers Effective for Time Series Forecasting?". It introduces a family of linear models (Linear, DLinear, NLinear)…LTSF-Linear is an official PyTorch implementation for "Are Transformers Effective for Time Series Forecasting?". It introduces a family of linear models (Linear, DLinear, NLinear)…

maml_rl

AI GitHub Repos

This repository provides code for reinforcement learning experiments detailed in the "Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks" paper. It enables…

Prototypical Networks

AI GitHub Repos

This repository contains the code for the NeurIPS 2017 paper "Prototypical Networks for Few-shot Learning." It provides an implementation of a novel approach to few-shot learning,…

LearningToCompare_FSL

AI GitHub Repos

This repository provides PyTorch code for the CVPR 2018 paper 'Learning to Compare: Relation Network for Few-Shot Learning'. It focuses on the Few-Shot Learning aspect, offering…

Model-Based Policy Optimization

AI GitHub Repos

This GitHub repository contains the code for the "When to Trust Your Model: Model-Based Policy Optimization" paper. It provides implementations for model-based policy optimization…

Handful of Trials

AI GitHub Repos

This repository contains experiment code for "Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models." It implements the PETS algorithm, combining…

Relational Graph Convolutional Networks

AI GitHub Repos

This project provides a Keras-based implementation of Relational Graph Convolutional Networks (R-GCNs). It is designed for semi-supervised node classification on directed…

CompGCN

AI GitHub Repos

PyTorch implementation of Composition-Based Multi-Relational Graph Convolutional Networks. It enables advanced graph representation learning for tasks like knowledge…CompGCN is a PyTorch implementation of Composition-Based Multi-Relational Graph Convolutional Networks. It enables advanced graph representation learning for tasks like knowledge…

GCNII

AI GitHub Repos

This repository provides a PyTorch implementation of the "Simple and Deep Graph Convolutional Networks" (GCNII) model. It includes code for replicating semi-supervised and…

PPNP & APPNP Models

AI GitHub Repos

This GitHub repository provides TensorFlow and PyTorch implementations for the PPNP and APPNP models, introduced in the "Predict then Propagate: Graph Neural Networks meet…

Simplifying Graph Convolutional Networks

AI GitHub Repos

This repository provides the official implementation for the "Simplifying Graph Convolutional Networks" paper. It offers a streamlined Graph Convolutional Network (GCN) model that…

Graph Neural PDEs

AI GitHub Repos

Open-source project providing Python code for Graph Neural Networks (GNNs) that treat graph learning as a continuous diffusion process. It implements GRAND…Graph Neural PDEs is an open-source project providing Python code for Graph Neural Networks (GNNs) that treat graph learning as a continuous diffusion process. It implements GRAND…

This blog post introduces a new PyTorch Hopfield layer based on modern Hopfield Networks. It generalizes classical Hopfield Networks to continuous states and patterns, offering…

OpenSpiel

AI GitHub Repos

Comprehensive framework for reinforcement learning research in games. It offers a collection of environments and algorithms for developing and testing general AI…OpenSpiel is a comprehensive framework for reinforcement learning research in games. It offers a collection of environments and algorithms for developing and testing general AI…

Temporal Graph Networks

AI GitHub Repos

Offer a generic, efficient framework for deep learning on dynamic graphs. This Python-based library utilizes novel memory modules and graph-based…Temporal Graph Networks (TGNs) offer a generic, efficient framework for deep learning on dynamic graphs. This Python-based library utilizes novel memory modules and graph-based…

xDeepFM

AI GitHub Repos

Open-source deep learning toolkit for factorization models, specifically releasing the source code for the eXtreme Deep Factorization Machine (xDeepFM) model. It…xDeepFM is an open-source deep learning toolkit for factorization models, specifically releasing the source code for the eXtreme Deep Factorization Machine (xDeepFM) model. It…

Attentional Factorization Machine

AI GitHub Repos

This GitHub repository provides a TensorFlow implementation of the Attentional Factorization Machine (AFM) model. It allows users to train and evaluate AFM for tasks like…

LightGCN

AI GitHub Repos

Simplified Graph Convolutional Network designed for recommendation systems. It focuses on the essential neighborhood aggregation component for collaborative…LightGCN is a simplified Graph Convolutional Network designed for recommendation systems. It focuses on the essential neighborhood aggregation component for collaborative…

Neural Graph Collaborative Filtering

AI GitHub Repos

Recommendation framework leveraging graph neural networks. It explicitly encodes collaborative signals through high-order…Neural Graph Collaborative Filtering (NGCF) is a recommendation framework leveraging graph neural networks. It explicitly encodes collaborative signals through high-order…

Intel® Neural Compressor

AI GitHub Repos

Open-source Python library offering popular model compression techniques for PyTorch, TensorFlow, and JAX. It supports advanced quantization for…Intel® Neural Compressor is an open-source Python library offering popular model compression techniques for PyTorch, TensorFlow, and JAX. It supports advanced quantization for…

AI-powered writing assistant designed to enhance productivity for researchers and academics. It integrates with your workflow to help generate, organize, and refine…Elephas is an AI-powered writing assistant designed to enhance productivity for researchers and academics. It integrates with your workflow to help generate, organize, and refine…

Open-source neural network framework written in Java. It simplifies the development and implementation of artificial neural networks for various applications. The…Neuroph is an open-source neural network framework written in Java. It simplifies the development and implementation of artificial neural networks for various applications. The…

StarCoder

AI GitHub Repos

Large language model trained on source code and natural language. It excels at code generation, completion, and text generation tasks. This repository provides…StarCoder is a large language model trained on source code and natural language. It excels at code generation, completion, and text generation tasks. This repository provides…

LLaVA

AI GitHub Repos

Visual instruction tuning model that combines large language and vision capabilities. It aims to achieve GPT-4V level performance, enabling multimodal understanding and…LLaVA is a visual instruction tuning model that combines large language and vision capabilities. It aims to achieve GPT-4V level performance, enabling multimodal understanding and…

Kandinsky 2

AI GitHub Repos

Multilingual text-to-image latent diffusion model. It offers advanced image generation capabilities, including text-to-image, image-to-image, and inpainting. The…Kandinsky 2 is a multilingual text-to-image latent diffusion model. It offers advanced image generation capabilities, including text-to-image, image-to-image, and inpainting. The…