AI Models
Explore audio synthesis demos powered by DiffWave, a versatile diffusion model. This resource showcases neural vocoding, class-conditional generation, unconditional waveform…
Discover the best Machine Learning & Data Science — from AI apps and models to mobile apps, frameworks, and platforms, this is where data scientists and machine learning engineers find tools that enhance their workflows: training models, deploying AI solutions, visualizing data, and optimizing algorithms. Browse 1,073 Machine Learning & Data Science AI Tools across AI Apps, AI Models, AI Mobile Apps, AI Frameworks and AI Platforms and more.
1,073 tools
AI Models
Explore audio synthesis demos powered by DiffWave, a versatile diffusion model. This resource showcases neural vocoding, class-conditional generation, unconditional waveform…
AI Models
FastSpeech 2 is an end-to-end text-to-speech model that enhances voice quality and training speed. It directly incorporates speech variation information like pitch and energy,…
AI Models
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…
AI Models
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…
AI Models
MUSE is a Python library for creating multilingual word embeddings, supporting both unsupervised and supervised methods. It aligns fastText embeddings into a common space and…
AI Models
A Hidden Markov Model (HMM) is a statistical model where observations depend on an unobservable Markov process. It's used to infer hidden states from observed data, finding…
AI Frameworks
LIBSVM is an integrated software package for support vector classification, regression, and distribution estimation. It offers a user-friendly interface, supports multi-class…
AI Models
An autoencoder is a type of neural network designed for unsupervised learning, focusing on learning efficient data encodings. It comprises an encoder that compresses data into a…
AI Models
The Vision Transformer (ViT) repository provides models and code for image recognition tasks. It includes implementations of the Vision Transformer and MLP-Mixer architectures,…
AI Models
FNet is an efficient Transformer-like encoder architecture that replaces self-attention with Fourier Transforms. Developed by Google Research, it offers a high-performance…
AI Models
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…
AI Models
LayoutLM is a multimodal pre-training model for visually-rich document understanding and information extraction. It combines text, layout, and image information to achieve…
AI Models
UniLM is a large-scale, self-supervised pre-training framework developed by Microsoft. It enables models to learn across diverse tasks, languages, and modalities, including text,…
AI Models
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…
AI Models
PEGASUS is a state-of-the-art abstractive text summarization model developed by Google Research. It utilizes a novel pre-training objective, gap-sentence generation, to achieve…
AI Models
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…
AI Models
DeBERTa is a large-scale pre-trained language model developed by Microsoft Research. It surpasses T5 11B models in performance and achieves human-level results on SuperGLUE…
AI Models
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…
AI Models
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…
AI Models
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…
AI Models
node2vec is a scalable framework for learning continuous feature representations of nodes in graphs. It optimizes neighborhood preservation through biased random walks, enabling…
This repository provides the official code implementation for the research paper "How Attentive are Graph Attention Networks?" published at ICLR'2022. It offers reproducible…
AI Models
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,…
AI Models
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…
AI Models
StyleSwin is a transformer-based Generative Adversarial Network (GAN) designed for high-resolution image generation. It leverages Swin transformers and a novel double attention…
AI Models
Decision Transformer reframes reinforcement learning as a sequence modeling problem, leveraging Transformer architectures like GPT-x and BERT. It generates optimal actions by…
AI Models
PyTorch implementation of the TabNet paper, offering an attentive and interpretable approach to tabular data learning. It supports classification, regression, and multi-task…
AI Models
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…
This GitHub repository provides the official implementation for the NeurIPS 2021 paper 'Revisiting Deep Learning Models for Tabular Data.' It explores deep learning architectures…
AI Models
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…
AI Models
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…
AI Models
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…
AI Models
Squeeze-and-Excitation Networks (SENet) is a deep learning model architecture that adaptively recalibrates channel-wise feature responses. It enhances performance by explicitly…
AI Models
The ShuffleNet Series repository offers a collection of highly efficient convolutional neural network models designed for mobile devices and resource-constrained environments. It…
AI Models
MnasNet automates the design of mobile machine learning models by incorporating speed constraints directly into the search reward function. This platform-aware approach identifies…
AI Models
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…
AI Models
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…
AI Models
MobileOne is an efficient neural network backbone designed for mobile devices. It optimizes for inference speed, achieving under 1 ms on an iPhone12 with high accuracy. This model…
AI Models
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,…
AI Models
Deformable Convolutional Networks (DCN) enhance deep learning models by adaptively learning sampling offsets for convolutional kernels. This allows networks to better capture…
AI Models
PointPillars for KITTI object detection is a GitHub repository providing code to reproduce results from the PointPillars paper on the KITTI dataset. It's a fork of the SECOND…
AI Models
SECOND PyTorch is an open-source implementation of the SECOND object detection network for KITTI and NuScenes datasets. It utilizes sparse convolutions for efficient 3D object…
AI Models
Panoptic FPN unifies instance and semantic segmentation into a single network architecture. It enhances Mask R-CNN with a semantic segmentation branch using a shared Feature…
AI Models
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…
AI Models
InferSent is a sentence embedding method that generates semantic representations for English sentences. Trained on natural language inference data, it generalizes well to various…
AI Models
CTRL is a 1.6 billion-parameter conditional transformer language model for controllable text generation. It conditions on control codes to specify domain, entities, and…
AI Models
The BAAI Models platform offers a diverse collection of AI models for natural language processing, multimodal understanding, and protein analysis. Explore and download models for…
AI Models
DialoGPT is a large-scale pretrained language model for dialogue response generation. Developed by Microsoft, it leverages GPT-2 architecture and is trained on extensive Reddit…
AI Models
Meena is a 2.6 billion parameter neural conversational model designed for open-domain dialogue. It aims to provide more sensible and specific responses than existing chatbots,…
AI Models
GODEL is a large-scale pre-trained Transformer-based model for goal-directed dialog generation. It excels at response generation grounded in external text, enabling efficient…
AI Models
RETRO (Retrieval Enhanced Transformers) is an AI model that augments transformer architectures with retrieval capabilities. It accesses a vast database of text passages, including…
AI Models
BERT offers two multilingual models: Cased and Uncased, supporting over 100 languages. The Cased model is recommended for non-Latin alphabets and general use, while the Uncased…
AI Models
LS-GAN, or Loss-Sensitive Generative Adversarial Networks, is a project focused on advancing GANs. It introduces a novel approach to loss functions, aiming for improved generation…
AI Models
This GitHub repository provides an official Chainer implementation for conditional image generation. It utilizes spectral normalization and a projection discriminator for…
AI Models
This repository provides a PyTorch implementation of MADE (Masked Autoencoder Density Estimation). It enables turning auto-encoders into autoregressive density models by masking…
AI Models
Lyra is a novel, very low-bitrate speech codec developed by Google. It leverages machine learning to compress voice signals, enabling high-quality audio communication even on the…
AI Models
SoundStorm is an AI model for efficient, non-autoregressive audio generation. It produces high-quality audio two orders of magnitude faster than previous methods, maintaining…
AI Models
GluonTS is a Python package for probabilistic time series modeling, leveraging deep learning models built on PyTorch and MXNet. It facilitates accurate forecasting by generating…
AI Models
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…
AI Models
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…
AI Models
The Informer2020 GitHub repository provides the PyTorch implementation for the Informer model, designed for efficient long sequence time-series forecasting. It features ProbSparse…
AI Models
FEDformer is a Frequency Enhanced Decomposed Transformer designed for efficient long-term time series forecasting. It achieves linear complexity with sequence length,…
AI Models
PatchTST is an official implementation of a Transformer-based model for long-term time series forecasting. It segments time series into patches and processes them independently,…
AI Models
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)…
AI Models
TimesNet is a foundation model for general time series analysis, transforming 1D series into 2D space for enhanced modeling. It achieves state-of-the-art results in forecasting,…
AI Models
This repository provides code for reinforcement learning experiments detailed in the "Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks" paper. It enables…
AI Models
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,…
AI Models
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…
This GitHub repository contains the code for the paper "Generative Adversarial Imitation Learning." It implements Trust Region Policy Optimization and provides scripts for…
AI Models
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…
AI Models
This repository contains experiment code for "Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models." It implements the PETS algorithm, combining…
AI Models
PlaNet is a model-based reinforcement learning algorithm that plans from pixels by learning latent dynamics. It efficiently predicts future rewards in a learned latent space,…
AI Models
OpenAI Five is a team of five neural networks developed by OpenAI. This AI system has demonstrated the capability to compete against and defeat amateur human teams in the complex…
AI Models
AlphaStar is an AI model that achieved Grandmaster level in StarCraft II, a complex real-time strategy game. It utilizes multi-agent reinforcement learning and imitation learning,…
AI Models
Policy gradient methods are a class of reinforcement learning algorithms that directly learn a policy function. Unlike value-based methods, they optimize a policy's parameters to…
AI Models
SARSA is a reinforcement learning algorithm for learning Markov decision process policies. It updates Q-values based on the agent's current state, action, reward, next state, and…
AI Models
Q-learning is a model-free reinforcement learning algorithm that trains an agent to assign values to actions based on current states. It optimizes decision-making by maximizing…
This project provides a Keras-based implementation of Relational Graph Convolutional Networks (R-GCNs). It is designed for semi-supervised node classification on directed…
AI Models
CompGCN is a PyTorch implementation of Composition-Based Multi-Relational Graph Convolutional Networks. It enables advanced graph representation learning for tasks like knowledge…
AI Models
This repository provides a PyTorch implementation of the "Simple and Deep Graph Convolutional Networks" (GCNII) model. It includes code for replicating semi-supervised and…
AI Models
This GitHub repository provides TensorFlow and PyTorch implementations for the PPNP and APPNP models, introduced in the "Predict then Propagate: Graph Neural Networks meet…
This repository provides the official implementation for the "Simplifying Graph Convolutional Networks" paper. It offers a streamlined Graph Convolutional Network (GCN) model that…
AI Models
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…
AI Models
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…
AI Models
A self-organizing map (SOM) is an unsupervised machine learning technique that reduces data dimensionality while preserving topological structure. It creates a low-dimensional…
AI Models
A radial basis function network is an artificial neural network utilizing radial basis functions as activation functions. It's adept at function approximation, time series…
AI Models
A Bayesian network is a probabilistic graphical model representing variables and their conditional dependencies using a directed acyclic graph. It's ideal for predicting the…
AI Models
Markov Random Fields (MRFs) are undirected graphical models representing probabilistic relationships between random variables. They are crucial in AI for modeling dependencies in…
AI Models
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…
AI Models
SAGPool is an official PyTorch implementation of Self-Attention Graph Pooling, presented at ICML 2019. This read-only, archived repository provides the code for a novel graph…
Scikit-learn's semi-supervised learning module enables models to leverage unlabeled data alongside labeled data for improved generalization. It offers algorithms like Self…
AI Models
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…
AI Models
EvolveGCN provides code for Evolving Graph Convolutional Networks designed for dynamic graphs. This AI model, published in AAAI 2020, offers a framework for analyzing and modeling…
AI Models
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…
AI Models
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…
AI Models
TensorFlow Models is a GitHub repository offering a collection of models and examples built with TensorFlow. It serves as a central hub for developers to access, contribute to,…
AI Models
LightGCN is a simplified Graph Convolutional Network designed for recommendation systems. It focuses on the essential neighborhood aggregation component for collaborative…
AI Models
Neural Graph Collaborative Filtering (NGCF) is a recommendation framework leveraging graph neural networks. It explicitly encodes collaborative signals through high-order…
AI Models
This AI model introduces Poincaré embeddings for learning hierarchical representations of symbolic data. It embeds data into hyperbolic space, capturing both hierarchy and…
AI Frameworks
TensorFlow is an open-source framework designed for machine learning, enabling beginners and experts to build and deploy models across various platforms including desktop, mobile,…
AI Frameworks
PyTorch documentation provides comprehensive resources for developers using this optimized tensor library for deep learning. It covers stable and unstable features, user guides,…
AI Frameworks
Keras is a Python deep learning framework. It provides a high-level API for building and training neural networks. Keras focuses on enabling fast experimentation and…
AI Frameworks
scikit-learn is a Python library offering simple and efficient tools for predictive data analysis. Built on NumPy, SciPy, and Matplotlib, it provides accessible and reusable…
AI Frameworks
Apache MXNet is an open-source deep learning framework designed for flexibility and efficiency. It supports hybrid front-ends for seamless transitions between eager and symbolic…
AI Frameworks
The Microsoft Cognitive Toolkit (CNTK) is an open-source deep learning framework for commercial-grade distributed training. It describes neural networks using directed graphs,…
AI Frameworks
Eclipse Deeplearning4j is an open-source, distributed deep learning framework for the JVM. It offers a comprehensive ecosystem for building, training, and deploying neural…
AI Frameworks
Caffe is a deep learning framework developed by Berkeley AI Research, prioritizing expression, speed, and modularity. It allows users to define models and optimizations through…
AI Frameworks
OpenNN is a free and open-source software library for neural networks. It provides a comprehensive set of tools for developing and implementing artificial intelligence models.…
AI Frameworks
XGBoost is an optimized, distributed gradient boosting library designed for efficiency, flexibility, and portability. It implements machine learning algorithms within the Gradient…
AI Frameworks
LightGBM is a high-performance gradient boosting framework utilizing tree-based learning algorithms. It is engineered for distributed and efficient operation, offering faster…
AI Frameworks
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 Frameworks
Hugging Face Transformers is an open-source framework centralizing state-of-the-art machine learning model definitions for text, vision, audio, and multimodal tasks. It ensures…
AI Frameworks
Stanza is a Python natural language processing library offering accurate and efficient tools for linguistic analysis across many human languages. It provides a neural network…
AI Frameworks
OpenCV Modules provides comprehensive documentation for the OpenCV library's various modules. It serves as a central resource for developers to understand and utilize the…
AI Frameworks
Detectron2 is an open-source object detection and segmentation library developed by Facebook AI Research. It provides a flexible framework for building and training…
AI Frameworks
Ultralytics YOLO is a leading AI framework for real-time object detection and image segmentation. It offers advanced models like YOLO26, supporting tasks such as detection,…
AI Frameworks
OpenVINO is an open-source toolkit for deploying high-performance AI solutions across diverse hardware. It enables developers to convert, optimize, and run inference for both…
AI Frameworks
Gymnasium provides a standard API for reinforcement learning and a diverse set of reference environments. It is a maintained fork of OpenAI's Gym library, offering a simple,…
AI Frameworks
Stable-Baselines3 (SB3) offers reliable reinforcement learning algorithm implementations in PyTorch. It provides a unified structure, PEP 8 compliance, extensive documentation,…
AI Frameworks
docs.ray.io is the official documentation portal for Ray, an open-source framework designed for scaling AI and Python applications. It provides comprehensive guides, API…
AI Frameworks
TensorFlow Agents is a library for reinforcement learning within TensorFlow. It simplifies the design, implementation, and testing of new reinforcement learning algorithms by…
AI Frameworks
Amazon SageMaker AI provides a fully managed service for machine learning. Data scientists and developers can build, train, and deploy ML models efficiently. It offers tools for…
AI Frameworks
IBM Watson on IBM Cloud provides AI services and documentation for developers. Explore resources like SDKs, tutorials, and API references to integrate advanced AI capabilities…
AI Frameworks
H2O.ai Documentation provides comprehensive resources for its suite of AI and machine learning platforms. Explore guides for H2O AI Cloud, Generative AI tools like h2oGPT, H2O…
AI Frameworks
Databricks documentation offers how-to guides and reference information for data analysts, scientists, and engineers. It covers the Databricks Data Intelligence Platform, enabling…
AI Frameworks
Weights & Biases offers a platform for developing AI models and shipping LLM applications. It provides experiment tracking, evaluation, and observability tools, alongside managed…
AI Frameworks
SHAP (SHapley Additive exPlanations) is a game theoretic approach to explain machine learning model outputs. It connects optimal credit allocation with local explanations using…
LIME is a Python package for generating local, interpretable, and model-agnostic explanations of machine learning model predictions. It helps understand why a model makes a…
AI Frameworks
AI Fairness 360 (AIF360) is an open-source toolkit designed to help researchers and developers detect, understand, and mitigate unwanted bias in machine learning models. It…
AI Frameworks
Fairlearn is an open-source toolkit designed to assess and improve the fairness of AI systems. It provides tools for identifying and mitigating unfairness in machine learning…
AI Frameworks
LlamaIndex is a leading framework for building LLM-powered agents and applications over your data. It simplifies data ingestion, indexing, and querying, enabling complex workflows…
AI Frameworks
Cohere's comprehensive API documentation empowers developers to seamlessly integrate advanced natural language processing and generation capabilities into their applications.…
AI Frameworks
ONNX (Open Neural Network Exchange) provides documentation for its 1.23.0 version. It covers core concepts, Python integration, API references, versioning, data structures, and…
AI Frameworks
Turi Create is an AI framework that simplifies the development of custom machine learning models. It allows users to add features like recommendations, object detection, and image…
AI Frameworks
DeepSpeed is a deep learning optimization library designed to simplify and enhance the efficiency of distributed training for AI models. It focuses on making large-scale model…
AI Frameworks
MindSpore provides a comprehensive set of APIs for building, training, and inferring AI models. It offers interfaces for framework foundations, operations, neural network layers,…
AI Frameworks
Chainer is a flexible, intuitive deep learning framework supporting CUDA computation for GPU acceleration. It allows easy multi-GPU use and handles diverse network architectures.…
AI Frameworks
Orfeo ToolBox (OTB) is an open-source framework for advanced remote sensing image processing. It handles large-scale optical, multispectral, and radar data. OTB offers diverse…
AI Frameworks
Vowpal Wabbit documentation provides comprehensive resources for its machine learning framework. It includes generated documentation for Python bindings and C++ implementations,…
AI Frameworks
Mallet is a Java-based package for statistical natural language processing and machine learning applications to text. It offers tools for document classification, clustering,…
AI Frameworks
LIBLINEAR is a high-performance C/C++ library for large-scale linear classification and regression. It efficiently handles datasets with millions of instances and features,…
AI Frameworks
Intel Gaudi documentation provides comprehensive resources for Intel Gaudi AI accelerators, including Gaudi 2 and Gaudi 3. It offers guidance on model migration, code samples,…
AI Frameworks
LiteRT is Google's high-performance on-device machine learning framework for deploying GenAI and ML models on edge platforms. It offers efficient conversion, runtime, and…
AI Frameworks
Core ML is an Apple framework that allows developers to integrate machine learning models directly into their apps. It provides a unified representation for models, enabling…
AI Frameworks
ML Kit is a free mobile SDK for Android and iOS that integrates Google's machine learning expertise into apps. It offers easy-to-use Vision and Natural Language APIs for…
AI Frameworks
TensorFlow Extended (TFX) is a Google-production-scale machine learning platform. It provides a configuration framework and shared libraries to integrate common components for…
AI Frameworks
BigDL-LLM is an open-source library for running large language models (LLMs) on Intel XPU hardware, from laptops to cloud GPUs. It supports INT4/FP4/INT8/FP8 quantization for low…
AI Frameworks
Dask-ML offers scalable machine learning in Python, integrating with libraries like Scikit-Learn and XGBoost. It addresses challenges of large models and datasets exceeding RAM by…
AI Frameworks
AutoKeras is an open-source AutoML system built on Keras, designed to make machine learning accessible. It automates the process of finding the best neural network architecture…
AI Frameworks
auto-sklearn is an automated machine learning toolkit that acts as a drop-in replacement for scikit-learn estimators. It automates algorithm selection and hyperparameter tuning,…
ToolPotion is the AI tools directory people browse to find the best Machine Learning & Data Science. Listing is free and every submission is editor-reviewed — put your tool where the search starts.