Project Awesome project awesome

Machine Learning

Collection 72.1k stars GitHub

Contents

Machine Learning

neonrvm 40 (archived)

neonrvm is an open source machine learning library based on RVM technique. It's written in C programming language and comes with Python programming language bindings.

Caffe 34.8k updated 1y ago

A deep learning framework developed with cleanliness, readability, and speed in mind.

Shogun 3.1k updated 2y ago

The Shogun Machine Learning Toolbox.

XGBoost 28.2k updated 3d ago

Python bindings for eXtreme Gradient Boosting (Tree) Library.

xLearn 3.1k updated 2y ago

A high performance, easy-to-use, and scalable machine learning package, which can be used to solve large-scale machine learning problems. xLearn is especially useful for solving machine learning problems on large-scale sparse data, which is very common in Internet services such as online advertisement and recommender systems.

auto_ml 1.7k updated 5y ago

Automated machine learning for production and analytics. Lets you focus on the fun parts of ML, while outputting production-ready code, and detailed analytics of your dataset and results. Includes support for NLP, XGBoost, CatBoost, LightGBM, and soon, deep learning.

pattern 8.9k updated 1y ago

Web mining module for Python.

dtaidistance 1.2k updated 1mo ago

High performance library for time series distances (DTW) and time series clustering.

einops 9.4k updated 1mo ago

Deep learning operations reinvented (for pytorch, tensorflow, jax and others).

machine learning

automated build consisting of a web-interface, and set of programmatic-interface API, for support vector machines. Corresponding dataset(s) are stored into a SQL database, then generated model(s) used for prediction(s), are stored into a NoSQL datastore.

InterpretML 6.8k updated 2d ago

InterpretML implements the Explainable Boosting Machine (EBM), a modern, fully interpretable machine learning model based on Generalized Additive Models (GAMs). This open-source package also provides visualization tools for EBMs, other glass-box models, and black-box explanations.

ChefBoost 486 updated 8mo ago

a lightweight decision tree framework for Python with categorical feature support covering regular decision tree algorithms such as ID3, C4.5, CART, CHAID and regression tree; also some advanced bagging and boosting techniques such as gradient boosting, random forest and adaboost.

Bayesian Methods for Hackers 28.5k updated 1y ago

Book/iPython notebooks on Probabilistic Programming in Python.

Featureforge 385 updated 8y ago

A set of tools for creating and testing machine learning features, with a scikit-learn compatible API.

metric-learn 1.4k updated 6d ago

A Python module for metric learning.

MCP Memory Service 1.5k updated 2d ago

Universal memory service with semantic search, autonomous consolidation, and multi-client support for AI applications.

OpenMetricLearning 985 updated 3mo ago

A PyTorch-based framework to train and validate the models producing high-quality embeddings.

Intel(R) Extension for Scikit-learn 1.3k updated 2d ago

A seamless way to speed up your Scikit-learn applications with no accuracy loss and code changes.

SimpleAI 988 updated 2y ago

Python implementation of many of the artificial intelligence algorithms described in the book "Artificial Intelligence, a Modern Approach". It focuses on providing an easy to use, well documented and tested library.

Neurolink 120 updated 2d ago

Enterprise-grade LLM integration framework for building production-ready AI applications with built-in hallucination prevention, RAG, and MCP support.

Pylearn2 2.8k updated 4y ago

A Machine Learning library based on Theano. [Deprecated]

keras 63.9k updated yesterday

High-level neural networks frontend for TensorFlow, CNTK and Theano.

Lasagne 3.9k updated 4y ago

Lightweight library to build and train neural networks in Theano.

hebel 1.2k updated 5y ago

GPU-Accelerated Deep Learning Library in Python. [Deprecated]

Chainer 5.9k updated 2y ago

Flexible neural network framework.

skforecast 1.5k updated yesterday

Python library for time series forecasting using machine learning models. It works with any regressor compatible with the scikit-learn API, including popular options like LightGBM, XGBoost, CatBoost, Keras, and many others.

Feature-engine 2.2k updated 17d ago

Open source library with an exhaustive battery of feature engineering and selection methods based on pandas and scikit-learn.

gensim 16.4k updated 4mo ago

Topic Modelling for Humans.

Gower Express

The Fastest Gower Distance Implementation for Python. GPU-accelerated similarity matching for mixed data types, 15-25% faster than alternatives with production-ready reliability.

topicwizard 145 updated 1y ago

Interactive topic model visualization/interpretation framework.

topik 92 (archived)

Topic modelling toolkit. [Deprecated]

PyBrain 2.9k updated 1y ago

Another Python Machine Learning Library.

Brainstorm

Fast, flexible and fun neural networks. This is the successor of PyBrain.

Crab 1.2k updated 5y ago

A flexible, fast recommender engine. [Deprecated]

python-recsys 1.5k updated 5y ago

A Python library for implementing a Recommender System.

Image-to-Image Translation with Conditional Adversarial Networks 143 updated 9y ago

Implementation of image to image (pix2pix) translation from the paper by isola et al.[DEEP LEARNING]

Restricted Boltzmann Machines

Restricted Boltzmann Machines in Python. [DEEP LEARNING]

Bolt 87 updated 14y ago

Bolt Online Learning Toolbox. [Deprecated]

CoverTree 31 updated 14y ago

Python implementation of cover trees, near-drop-in replacement for scipy.spatial.kdtree [Deprecated]

nilearn 1.4k updated yesterday

Machine learning for NeuroImaging in Python.

neuropredict 104 updated 2mo ago

Aimed at novice machine learners and non-expert programmers, this package offers easy (no coding needed) and comprehensive machine learning (evaluation and full report of predictive performance WITHOUT requiring you to code) in Python for NeuroImaging and any other type of features. This is aimed at absorbing much of the ML workflow, unlike other packages like nilearn and pymvpa, which require you to learn their API and code to produce anything useful.

Pyevolve 315 updated 4y ago

Genetic algorithm framework. [Deprecated]

breze 95 updated 9y ago

Theano based library for deep and recurrent neural networks.

Cortex 8.0k updated 1y ago

Open source platform for deploying machine learning models in production.

pyhsmm 575 updated 1y ago

library for approximate unsupervised inference in Bayesian Hidden Markov Models (HMMs) and explicit-duration Hidden semi-Markov Models (HSMMs), focusing on the Bayesian Nonparametric extensions, the HDP-HMM and HDP-HSMM, mostly with weak-limit approximations.

SKLL 560 updated 9mo ago

A wrapper around scikit-learn that makes it simpler to conduct experiments.

Spearmint

Spearmint is a package to perform Bayesian optimization according to the algorithms outlined in the paper: Practical Bayesian Optimization of Machine Learning Algorithms. Jasper Snoek, Hugo Larochelle and Ryan P. Adams. Advances in Neural Information Processing Systems, 2012. [Deprecated]

Pebl 104 updated 14y ago

Python Environment for Bayesian Learning. [Deprecated]

Theano 10.0k updated 2y ago

Optimizing GPU-meta-programming code generating array oriented optimizing math compiler in Python.

TensorFlow 194.3k updated 2d ago

Open source software library for numerical computation using data flow graphs.

pomegranate

Hidden Markov Models for Python, implemented in Cython for speed and efficiency.

python-timbl 18 updated 10mo ago

A Python extension module wrapping the full TiMBL C++ programming interface. Timbl is an elaborate k-Nearest Neighbours machine learning toolkit.

deap 6.4k updated 4mo ago

Evolutionary algorithm framework.

pydeep 1.4k updated 5y ago

Deep Learning In Python. [Deprecated]

mlxtend 5.1k updated 2mo ago

A library consisting of useful tools for data science and machine learning tasks.

neon 3.9k (archived)

Nervana's high-performance Python-based Deep Learning framework [DEEP LEARNING]. [Deprecated]

ML-From-Scratch 31.1k updated 2y ago

Implementations of Machine Learning models from scratch in Python with a focus on transparency. Aims to showcase the nuts and bolts of ML in an accessible way.

xRBM 55 updated 8y ago

A library for Restricted Boltzmann Machine (RBM) and its conditional variants in Tensorflow.

stacked_generalization 119 updated 7y ago

Implementation of machine learning stacking technique as a handy library in Python.

modAL 2.3k updated 2y ago

A modular active learning framework for Python, built on top of scikit-learn.

Cogitare 77 (archived)

A Modern, Fast, and Modular Deep Learning and Machine Learning framework for Python.

Parris 314 updated 1mo ago

Parris, the automated infrastructure setup tool for machine learning algorithms.

Turi Create 11.2k (archived)

Machine learning from Apple. Turi Create simplifies the development of custom machine learning models. You don't have to be a machine learning expert to add recommendations, object detection, image classification, image similarity or activity classification to your app.

mlens 863 updated 2y ago

A high performance, memory efficient, maximally parallelized ensemble learning, integrated with scikit-learn.

Thampi 9 (archived)

Machine Learning Prediction System on AWS Lambda

MindsDB

Open Source framework to streamline use of neural networks.

Microsoft Recommenders 21.5k updated 3d ago

Examples and best practices for building recommendation systems, provided as Jupyter notebooks. The repo contains some of the latest state of the art algorithms from Microsoft Research as well as from other companies and institutions.

StellarGraph 3.0k updated 1y ago

Machine Learning on Graphs, a Python library for machine learning on graph-structured (network-structured) data.

BentoML 8.5k updated yesterday

Toolkit for package and deploy machine learning models for serving in production

MiraiML 26 updated 6y ago

An asynchronous engine for continuous & autonomous machine learning, built for real-time usage.

imbalanced-ensemble 419 updated 21d ago

Python toolbox for quick implementation, modification, evaluation, and visualization of ensemble learning algorithms for class-imbalanced data. Supports out-of-the-box multi-class imbalanced (long-tailed) classification.

C++

General-Purpose Machine Learning

Agentic Context Engine

In-context learning framework that allows agents to learn from execution feedback.

Speedster 8.4k updated 1y ago

Automatically apply SOTA optimization techniques to achieve the maximum inference speed-up on your hardware. [DEEP LEARNING]

DeepDetect 2.5k updated 2d ago

A machine learning API and server written in C++11. It makes state of the art machine learning easy to work with and integrate into existing applications.

DSSTNE 4.4k (archived)

A software library created by Amazon for training and deploying deep neural networks using GPUs which emphasizes speed and scale over experimental flexibility.

DyNet 3.4k updated 2y ago

A dynamic neural network library working well with networks that have dynamic structures that change for every training instance. Written in C++ with bindings in Python.

Fido 462 updated 6y ago

A highly-modular C++ machine learning library for embedded electronics and robotics.

FlexML 28 updated 2mo ago

Easy-to-use and flexible AutoML library for Python.

Intel oneAPI Data Analytics Library 645 updated 5d ago

A high performance software library developed by Intel and optimized for Intel's architectures. Library provides algorithmic building blocks for all stages of data analytics and allows to process data in batch, online and distributed modes.

LightGBM 18.2k updated 2d ago

Microsoft's fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

BanditLib 140 updated 2y ago

A simple Multi-armed Bandit library. [Deprecated]

MCGrad 19 updated 2d ago

A production-ready library for multicalibration, fairness, and bias correction in machine learning models.

N2D2 158 updated 1y ago

CEA-List's CAD framework for designing and simulating Deep Neural Network, and building full DNN-based applications on embedded platforms

oneDNN 4.0k updated yesterday

An open-source cross-platform performance library for deep learning applications.

proNet-core 3 updated 6y ago

A general-purpose network embedding framework: pair-wise representations optimization Network Edit.

PyCaret 9.7k updated 11mo ago

An open-source, low-code machine learning library in Python that automates machine learning workflows.

Vowpal Wabbit (VW)

A fast out-of-core learning system.

Warp-CTC 4.1k updated 2y ago

A fast parallel implementation of Connectionist Temporal Classification (CTC), on both CPU and GPU.

ThunderGBM 711 updated 1y ago

A fast library for GBDTs and Random Forests on GPUs.

LKYDeepNN 50 updated 5y ago

A header-only C++11 Neural Network library. Low dependency, native traditional chinese document.

Featuretools 7.6k updated 1mo ago

A library for automated feature engineering. It excels at transforming transactional and relational datasets into feature matrices for machine learning using reusable feature engineering "primitives".

skynet 62 updated 4y ago

A library for learning neural networks, has C-interface, net set in JSON. Written in C++ with bindings in Python, C++ and C#.

Feast 6.8k updated 2d ago

A feature store for the management, discovery, and access of machine learning features. Feast provides a consistent view of feature data for both model training and model serving.

Hopsworks

A data-intensive platform for AI with the industry's first open-source feature store. The Hopsworks Feature Store provides both a feature warehouse for training and batch based on Apache Hive and a feature serving database, based on MySQL Cluster, for online applications.

Polyaxon

A platform for reproducible and scalable machine learning and deep learning.

XAD 411 updated 23d ago

Comprehensive backpropagation tool for C++.

nndeploy 1.8k updated 11d ago

An Easy-to-Use and High-Performance AI deployment framework.

Python

Computer Vision

Openpose 33.9k updated 1y ago

A real-time multi-person keypoint detection library for body, face, hands, and foot estimation

LightlyTrain 1.4k updated 2d ago

Pretrain computer vision models on unlabeled data for industrial applications

Scikit-Image 6.5k updated yesterday

A collection of algorithms for image processing in Python.

Scikit-Opt 6.4k updated 6mo ago

Swarm Intelligence in Python (Genetic Algorithm, Particle Swarm Optimization, Simulated Annealing, Ant Colony Algorithm, Immune Algorithm, Artificial Fish Swarm Algorithm in Python)

PCV 2.0k updated 5y ago

Open source Python module for computer vision. [Deprecated]

face_recognition 56.2k updated 1y ago

Face recognition library that recognizes and manipulates faces from Python or from the command line.

deepface 22.4k updated 24d ago

A lightweight face recognition and facial attribute analysis (age, gender, emotion and race) framework for Python covering cutting-edge models such as VGG-Face, FaceNet, OpenFace, DeepFace, DeepID, Dlib and ArcFace.

retinaface

deep learning based cutting-edge facial detector for Python coming with facial landmarks

dockerface 191 updated 5y ago

Easy to install and use deep learning Faster R-CNN face detection for images and video in a docker container. [Deprecated]

Detectron 26.4k (archived)

FAIR's software system that implements state-of-the-art object detection algorithms, including Mask R-CNN. It is written in Python and powered by the Caffe2 deep learning framework. [Deprecated]

detectron2 34.3k updated 4d ago

FAIR's next-generation research platform for object detection and segmentation. It is a ground-up rewrite of the previous version, Detectron, and is powered by the PyTorch deep learning framework.

albumentations 15.3k (archived)

А fast and framework agnostic image augmentation library that implements a diverse set of augmentation techniques. Supports classification, segmentation, detection out of the box. Was used to win a number of Deep Learning competitions at Kaggle, Topcoder and those that were a part of the CVPR workshops.

pytessarct 6.3k updated 9d ago

Python-tesseract is an optical character recognition (OCR) tool for python. That is, it will recognize and "read" the text embedded in images. Python-tesseract is a wrapper for Google's Tesseract-OCR Engine.

imutils 4.6k updated 1y ago

A library containing Convenience functions to make basic image processing operations such as translation, rotation, resizing, skeletonization, and displaying Matplotlib images easier with OpenCV and Python.

PyTorchCV 53 updated 7y ago

A PyTorch-Based Framework for Deep Learning in Computer Vision.

joliGEN 280 updated 8d ago

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications.

neural-style-pt 863 updated 3y ago

A PyTorch implementation of Justin Johnson's neural-style (neural style transfer).

Detecto 625 updated 1y ago

Train and run a computer vision model with 5-10 lines of code.

neural-dream 149 updated 4y ago

A PyTorch implementation of DeepDream.

Deep High-Resolution-Net 4.5k updated 1y ago

A PyTorch implementation of CVPR2019 paper "Deep High-Resolution Representation Learning for Human Pose Estimation"

TF-GAN 966 updated 1y ago

TF-GAN is a lightweight library for training and evaluating Generative Adversarial Networks (GANs).

dream-creator 72 updated 3y ago

A PyTorch implementation of DeepDream. Allows individuals to quickly and easily train their own custom GoogleNet models with custom datasets for DeepDream.

Lucent 653 updated 1y ago

Tensorflow and OpenAI Clarity's Lucid adapted for PyTorch.

lightly 3.7k updated 2d ago

Lightly is a computer vision framework for self-supervised learning.

Learnergy 68 updated 1mo ago

Energy-based machine learning models built upon PyTorch.

IoT Owl 9 updated 4y ago

Light face detection and recognition system with huge possibilities, based on Microsoft Face API and TensorFlow made for small IoT devices like raspberry pi.

Exadel CompreFace

face recognition system that can be easily integrated into any system without prior machine learning skills. CompreFace provides REST API for face recognition, face verification, face detection, face mask detection, landmark detection, age, and gender recognition and is easily deployed with docker.

computer-vision-in-action 2.8k updated 1y ago

as known as `L0CV`, is a new generation of computer vision open source online learning media, a cross-platform interactive learning framework integrating graphics, source code and HTML. the L0CV ecosystem — Notebook, Datasets, Source Code, and from Diving-in to Advanced — as well as the L0CV Hub.

timm

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more.

segmentation_models.pytorch 11.4k updated 2d ago

A PyTorch-based toolkit that offers pre-trained segmentation models for computer vision tasks. It simplifies the development of image segmentation applications by providing a collection of popular architecture implementations, such as UNet and PSPNet, along with pre-trained weights, making it easier for researchers and developers to achieve high-quality pixel-level object segmentation in images.

segmentation_models

A TensorFlow Keras-based toolkit that offers pre-trained segmentation models for computer vision tasks. It simplifies the development of image segmentation applications by providing a collection of popular architecture implementations, such as UNet and PSPNet, along with pre-trained weights, making it easier for researchers and developers to achieve high-quality pixel-level object segmentation in images.

MLX 24.7k updated 2d ago

MLX is an array framework for machine learning on Apple silicon, developed by Apple machine learning research.

General-Purpose Machine Learning

CNTK 17.6k (archived)

Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit. Documentation can be found here.

RexMex

A general purpose recommender metrics library for fair evaluation.

ChemicalX 773 updated 2y ago

A PyTorch based deep learning library for drug pair scoring

Shapley 224 updated 2mo ago

A data-driven framework to quantify the value of classifiers in a machine learning ensemble.

igel 3.1k updated 3mo ago

A delightful machine learning tool that allows you to train/fit, test and use models without writing code

ML Model building

A Repository Containing Classification, Clustering, Regression, Recommender Notebooks with illustration to make them.

ML/DL project template 1.3k (archived)
PyTorch Frame 773 updated 2d ago

A Modular Framework for Multi-Modal Tabular Learning.

PyTorch Geometric 23.6k updated 3d ago

Graph Neural Network Library for PyTorch.

PyTorch Geometric Temporal 3.0k updated 6mo ago

A temporal extension of PyTorch Geometric for dynamic graph representation learning.

Little Ball of Fur 713 updated 3mo ago

A graph sampling extension library for NetworkX with a Scikit-Learn like API.

Karate Club 2.3k updated 1y ago

An unsupervised machine learning extension library for NetworkX with a Scikit-Learn like API.

Auto_ViML

Automatically Build Variant Interpretable ML models fast! AutoViML is pronounced "auto vimal", is a comprehensive and scalable Python AutoML toolkit with imbalanced handling, ensembling, stacking and built-in feature selection. Featured in <a href="https://towardsdatascience.com/why-automl-is-an-essential-new-tool-for-data-scientists-2d9ab4e25e46?source=friendslink&sk=d03a0cc55c23deb497d546d6b9be0653">Medium article</a>.

PyOD

Python Outlier Detection, comprehensive and scalable Python toolkit for detecting outlying objects in multivariate data. Featured for Advanced models, including Neural Networks/Deep Learning and Outlier Ensembles.

steppy 136 (archived)

Lightweight, Python library for fast and reproducible machine learning experimentation. Introduces a very simple interface that enables clean machine learning pipeline design.

steppy-toolkit 22 (archived)

Curated collection of the neural networks, transformers and models that make your machine learning work faster and more effective.

Couler 942 updated 1y ago

Unified interface for constructing and managing machine learning workflows on different workflow engines, such as Argo Workflows, Tekton Pipelines, and Apache Airflow.

Aim 6.0k updated 2d ago

An easy-to-use & supercharged open-source AI metadata tracker.

Natural Language Processing

colibri-core 129 updated 1mo ago

Python binding to C++ library for extracting and working with basic linguistic constructions such as n-grams and skipgrams in a quick and memory-efficient way.

pkuseg-python

A better version of Jieba, developed by Peking University.

Quepy 1.3k updated 5y ago

A python framework to transform natural language questions to queries in a database query language.

YAlign 130 updated 9y ago

A sentence aligner, a friendly tool for extracting parallel sentences from comparable corpora. [Deprecated]

jieba 34.8k updated 1y ago

Chinese Words Segmentation Utilities.

SnowNLP 6.6k updated 6y ago

A library for processing Chinese text.

spammy 145 updated 6y ago

A library for email Spam filtering built on top of NLTK

loso 81 updated 15y ago

Another Chinese segmentation library. [Deprecated]

genius 234 updated 7y ago

A Chinese segment based on Conditional Random Field.

nut 119 updated 12y ago

Natural language Understanding Toolkit. [Deprecated]

Rosetta 207 updated 3y ago

Text processing tools and wrappers (e.g. Vowpal Wabbit)

PyNLPl 477 updated 2y ago

Python Natural Language Processing Library. General purpose NLP library for Python. Also contains some specific modules for parsing common NLP formats, most notably for FoLiA, but also ARPA language models, Moses phrasetables, GIZA++ alignments.

PySS3 349 updated 5mo ago

Python package that implements a novel white-box machine learning model for text classification, called SS3. Since SS3 has the ability to visually explain its rationale, this package also comes with easy-to-use interactive visualizations tools (online demos).

python-ucto 31 updated 1mo ago

Python binding to ucto (a unicode-aware rule-based tokenizer for various languages).

python-frog 49 updated 1mo ago

Python binding to Frog, an NLP suite for Dutch. (pos tagging, lemmatisation, dependency parsing, NER)

python-zpar 50 (archived)

Python bindings for ZPar, a statistical part-of-speech-tagger, constituency parser, and dependency parser for English.

spaCy 33.4k updated 2d ago

Industrial strength NLP with Python and Cython.

PyStanfordDependencies 69 updated 7y ago

Python interface for converting Penn Treebank trees to Stanford Dependencies.

Distance

Levenshtein and Hamming distance computation. [Deprecated]

Fuzzy Wuzzy 9.3k (archived)

Fuzzy String Matching in Python.

Neofuzz 51 updated 11mo ago

Blazing fast, lightweight and customizable fuzzy and semantic text search in Python with fuzzywuzzy/thefuzz compatible API.

jellyfish 2.2k updated 16d ago

a python library for doing approximate and phonetic matching of strings.

textacy

higher-level NLP built on Spacy.

stanford-corenlp-python 611 updated 8y ago

Python wrapper for Stanford CoreNLP [Deprecated]

CLTK 901 updated 1mo ago

The Classical Language Toolkit.

Rasa 21.1k updated 1mo ago

A "machine learning framework to automate text-and voice-based conversations."

yase 14 (archived)

Transcode sentence (or other sequence) to list of word vector.

Polyglot 2.4k updated 2y ago

Multilingual text (NLP) processing toolkit.

DrQA 4.5k (archived)

Reading Wikipedia to answer open-domain questions.

Dedupe 4.4k updated 8mo ago

A python library for accurate and scalable fuzzy matching, record deduplication and entity-resolution.

Snips NLU 4.0k updated 2y ago

Natural Language Understanding library for intent classification and entity extraction

NeuroNER 1.7k updated 3y ago

Named-entity recognition using neural networks providing state-of-the-art-results

DeepPavlov 7.0k updated 7mo ago

conversational AI library with many pre-trained Russian NLP models.

BigARTM

topic modelling platform.

NALP 23 updated 2mo ago

A Natural Adversarial Language Processing framework built over Tensorflow.

DL Translate 497 updated 1y ago

A deep learning-based translation library between 50 languages, built with transformers.

Haystack 24.6k updated 2d ago

A framework for building industrial-strength applications with Transformer models and LLMs.

CometLLM 18.5k updated yesterday

Track, log, visualize and evaluate your LLM prompts and prompt chains.

NobodyWho 760 updated 2d ago

The simplest way to run an LLM locally. Supports tool calling and grammar constrained sampling.

Transformers

A deep learning library containing thousands of pre-trained models on different tasks. The goto place for anything related to Large Language Models.

TextCL

Text preprocessing package for use in NLP tasks.

VeritasGraph 255 updated 3d ago

Enterprise-Grade Graph RAG for Secure, On-Premise AI with Verifiable Attribution.

Python Machine Learning

Opik 18.5k updated 2d ago

Evaluate, trace, test, and ship LLM applications across your dev and production lifecycles.

numpy-ML 16.3k updated 2y ago

Reference implementations of ML models written in numpy

Neuraxle 614 updated 1mo ago

A framework providing the right abstractions to ease research, development, and deployment of your ML pipelines.

Cornac 1.0k updated 3mo ago

A comparative framework for multimodal recommender systems with a focus on models leveraging auxiliary data.

JAX 35.2k updated yesterday

JAX is Autograd and XLA, brought together for high-performance machine learning research.

Fastai 27.9k updated 28d ago

High-level wrapper built on the top of Pytorch which supports vision, text, tabular data and collaborative filtering.

scikit-multiflow 795 updated 2y ago

A machine learning framework for multi-output/multi-label and stream data.

Lightwood 502 updated 1mo ago

A Pytorch based framework that breaks down machine learning problems into smaller blocks that can be glued together seamlessly with objective to build predictive models with one line of code.

bayeso 95 updated 2mo ago

A simple, but essential Bayesian optimization package, written in Python.

mljar-supervised 3.3k updated yesterday

An Automated Machine Learning (AutoML) python package for tabular data. It can handle: Binary Classification, MultiClass Classification and Regression. It provides explanations and markdown reports.

evostra

A fast Evolution Strategy implementation in Python.

Determined 3.2k updated 1y ago

Scalable deep learning training platform, including integrated support for distributed training, hyperparameter tuning, experiment tracking, and model management.

PyGrid

Peer-to-peer network of data owners and data scientists who can collectively train AI models using PySyft

sktime 9.7k updated 5d ago

A unified framework for machine learning with time series

OPFython 37 updated 1mo ago

A Python-inspired implementation of the Optimum-Path Forest classifier.

Opytimizer

Python-based meta-heuristic optimization techniques.

Gradio 42.1k updated 2d ago

A Python library for quickly creating and sharing demos of models. Debug models interactively in your browser, get feedback from collaborators, and generate public links without deploying anything.

Hub 9.0k updated 1mo ago

Fastest unstructured dataset management for TensorFlow/PyTorch. Stream & version-control data. Store even petabyte-scale data in a single numpy-like array on the cloud accessible on any machine. Visit activeloop.ai for more info.

Synthia 65 updated 2y ago

Multidimensional synthetic data generation in Python.

ByteHub 61 updated 4y ago

An easy-to-use, Python-based feature store. Optimized for time-series data.

Backprop 241 updated 4y ago

Backprop makes it simple to use, finetune, and deploy state-of-the-art ML models.

River 5.8k updated yesterday

A framework for general purpose online machine learning.

FEDOT 703 updated 2d ago

An AutoML framework for the automated design of composite modelling pipelines. It can handle classification, regression, and time series forecasting tasks on different types of data (including multi-modal datasets).

Sklearn-genetic-opt

An AutoML package for hyperparameters tuning using evolutionary algorithms, with built-in callbacks, plotting, remote logging and more.

Evidently 7.3k updated 6d ago

Interactive reports to analyze machine learning models during validation or production monitoring.

Streamlit 44.0k updated yesterday

Streamlit is an framework to create beautiful data apps in hours, not weeks.

Optuna 13.7k updated 3d ago

Optuna is an automatic hyperparameter optimization software framework, particularly designed for machine learning.

Deepchecks 4.0k updated 2mo ago

Validation & testing of machine learning models and data during model development, deployment, and production. This includes checks and suites related to various types of issues, such as model performance, data integrity, distribution mismatches, and more.

Shapash 3.2k updated 1mo ago

Shapash is a Python library that provides several types of visualization that display explicit labels that everyone can understand.

Eurybia 216 updated 2d ago

Eurybia monitors data and model drift over time and securizes model deployment with data validation.

Colossal-AI

An open-source deep learning system for large-scale model training and inference with high efficiency and low cost.

skrub 1.6k updated 2d ago

Skrub is a Python library that eases preprocessing and feature engineering for machine learning on dataframes.

Upgini 349 updated 5d ago

Free automated data & feature enrichment library for machine learning - automatically searches through thousands of ready-to-use features from public and community shared data sources and enriches your training dataset with only the accuracy improving features.

AutoML-Implementation-for-Static-and-Dynamic-Data-Analytics 628 updated 1y ago

A tutorial to help machine learning researchers to automatically obtain optimized machine learning models with the optimal learning performance on any specific task.

SKBEL 26 updated 1y ago

A Python library for Bayesian Evidential Learning (BEL) in order to estimate the uncertainty of a prediction.

cleanlab 11.4k updated 2mo ago

The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

AutoGluon 10.1k updated 2d ago

AutoML for Image, Text, Tabular, Time-Series, and MultiModal Data.

PyBroker 3.2k updated 21d ago

Algorithmic Trading with Machine Learning.

Frouros 252 updated 2d ago

Frouros is an open source Python library for drift detection in machine learning systems.

CometML 171 updated 6d ago

The best-in-class MLOps platform with experiment tracking, model production monitoring, a model registry, and data lineage from training straight through to production.

ClearML 6.6k updated yesterday

Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution.

Okrolearn 3 (archived)

A python machine learning library created to combine powefull data analasys features with tensors and machine learning components, while maintaining support for other libraries.

pyclugen 10 updated 7mo ago

Multidimensional cluster generation in Python.

mlforgex updated 2mo ago

Lightweight ML utility for automated training, evaluation, and prediction with CLI and Python API support.

Clojure

Go

General-Purpose Machine Learning

Spago 1.8k updated 11mo ago

Self-contained Machine Learning and Natural Language Processing library in Go.

birdland 46 updated 6y ago

A recommendation library in Go.

eaopt 907 updated 1y ago

An evolutionary optimization library.

leaves 471 updated 1y ago

A pure Go implementation of the prediction part of GBRTs, including XGBoost and LightGBM.

gobrain 566 updated 5y ago

Neural Networks written in Go.

go-featureprocessing 126 (archived)

Fast and convenient feature processing for low latency machine learning in Go.

go-mxnet-predictor 54 updated 7y ago

Go binding for MXNet cpredictapi to do inference with a pre-trained model.

go-ml-benchmarks 32 updated 1y ago

benchmarks of machine learning inference for Go.

go-ml-transpiler

An open source Go transpiler for machine learning models.

golearn 9.4k updated 2y ago

Machine learning for Go.

goml 1.6k updated 3y ago

Machine learning library written in pure Go.

gorgonia 5.9k updated 1y ago

Deep learning in Go.

goro 374 updated 2y ago

A high-level machine learning library in the vein of Keras.

gorse

An offline recommender system backend based on collaborative filtering written in Go.

therfoo 18 (archived)

An embedded deep learning library for Go.

neat

Plug-and-play, parallel Go framework for NeuroEvolution of Augmenting Topologies (NEAT). [Deprecated]

go-pr 68 updated 12y ago

Pattern recognition package in Go lang. [Deprecated]

go-ml 199 updated 9y ago

Linear / Logistic regression, Neural Networks, Collaborative Filtering and Gaussian Multivariate Distribution. [Deprecated]

GoNN 360 updated 10y ago

GoNN is an implementation of Neural Network in Go Language, which includes BPNN, RBF, PCN. [Deprecated]

bayesian 811 updated 3mo ago

Naive Bayesian Classification for Golang. [Deprecated]

go-galib 200 updated 10y ago

Genetic Algorithms library written in Go / Golang. [Deprecated]

Cloudforest 748 updated 4y ago

Ensembles of decision trees in Go/Golang. [Deprecated]

go-dnn 8 (archived)

Deep Neural Networks for Golang (powered by MXNet)

Java

IRIS

Cortical.io's FREE NLP, Retina API Analysis Tool (written in JavaFX!) - See the Tutorial Video.

ClearTK 133 updated 2y ago

ClearTK provides a framework for developing statistical natural language processing (NLP) components in Java and is built on top of Apache UIMA. [Deprecated]

NLP4J 155 updated 4y ago

The NLP4J project provides software and resources for natural language processing. The project started at the Center for Computational Language and EducAtion Research, and is currently developed by the Center for Language and Information Research at Emory University. [Deprecated]

CogcompNLP

This project collects a number of core libraries for Natural Language Processing (NLP) developed in the University of Illinois' Cognitive Computation Group, for example illinois-core-utilities which provides a set of NLP-friendly data structures and a number of NLP-related utilities that support writing NLP applications, running experiments, etc, illinois-edison a library for feature extraction from illinois-core-utilities data structures and many other packages.

aerosolve 4.8k updated 4mo ago

A machine learning library by Airbnb designed from the ground up to be human friendly.

Chips-n-Salsa 74 updated 26d ago

A Java library for genetic algorithms, evolutionary computation, and stochastic local search, with a focus on self-adaptation / self-tuning, as well as parallel execution.

Datumbox 1.1k updated 2y ago

Machine Learning framework for rapid development of Machine Learning and Statistical applications.

ELKI 830 updated yesterday

Java toolkit for data mining. (unsupervised: clustering, outlier detection etc.)

Encog 752 updated 3y ago

An advanced neural network and machine learning framework. Encog contains classes to create a wide variety of networks, as well as support classes to normalize and process data for these neural networks. Encog trainings using multithreaded resilient propagation. Encog can also make use of a GPU to further speed processing time. A GUI based workbench is also provided to help model and train neural networks.

H2O 7.5k updated yesterday

ML engine that supports distributed learning on Hadoop, Spark or your laptop via APIs in R, Python, Scala, REST/JSON.

htm.java 318 updated 4y ago

General Machine Learning library using Numenta’s Cortical Learning Algorithm.

jSciPy 18 updated 1mo ago

A Java port of SciPy's signal processing module, offering filters, transformations, and other scientific computing utilities.

liblinear-java 310 updated 1y ago

Java version of liblinear.

Mahout 2.2k updated 2d ago

Distributed machine learning.

ORYX 1.8k (archived)

Lambda Architecture Framework using Apache Spark and Apache Kafka with a specialization for real-time large-scale machine learning.

rapaio 77 updated 9d ago

statistics, data mining and machine learning toolbox in Java.

SystemML 1.1k updated 7d ago

flexible, scalable machine learning (ML) language.

knn-java-library 7 updated 9y ago

Just a simple implementation of K-Nearest Neighbors algorithm using with a bunch of similarity measures.

Hadoop

Hadoop/HDFS.

Onyx 2.0k (archived)

Distributed, masterless, high performance, fault tolerant data processing. Written entirely in Clojure.

Spark 43.0k updated yesterday

Spark is a fast and general engine for large-scale data processing.

Impala 34 updated 3y ago

Real-time Query for Hadoop.

Deeplearning4j

Scalable deep learning for industry with parallel GPUs.

deepjavalibrary/djl 4.8k updated 4d ago

Deep Java Library (DJL) is an open-source, high-level, engine-agnostic Java framework for deep learning, designed to be easy to get started with and simple to use for Java developers.

Scala

General-Purpose Machine Learning

Microsoft ML for Apache Spark 5.2k updated yesterday

A distributed machine learning framework Apache Spark

ONNX-Scala 143 updated 1mo ago

An ONNX (Open Neural Network eXchange) API and backend for typeful, functional deep learning in Scala (3).

Conjecture

Scalable Machine Learning in Scalding.

brushfire

Distributed decision tree ensemble learning in Scala.

ganitha 109 (archived)

Scalding powered machine learning. [Deprecated]

adam 1.0k updated 8d ago

A genomics processing engine and specialized file format built using Apache Avro, Apache Spark and Parquet. Apache 2 licensed.

bioscala 115 updated 7mo ago

Bioinformatics for the Scala programming language

BIDMach 919 updated 3y ago

CPU and GPU-accelerated Machine Learning Library.

Figaro 759 updated 3y ago

a Scala library for constructing probabilistic models.

H2O Sparkling Water 977 updated 4mo ago

H2O and Spark interoperability.

Saul 64 updated 6y ago

Flexible Declarative Learning-Based Programming.

SwiftLearner 40 updated 1y ago

Simply written algorithms to help study ML or write your own implementations.

doddle-model

An in-memory machine learning library built on top of Breeze. It provides immutable objects and exposes its functionality through a scikit-learn-like API.

TensorFlow Scala 940 updated 3y ago

Strongly-typed Scala API for TensorFlow.

isolation-forest 253 updated 2d ago

A distributed Spark/Scala implementation of the isolation forest algorithm for unsupervised outlier detection, featuring support for scalable training and ONNX export for easy cross-platform inference.

JavaScript

General-Purpose Machine Learning

Catniff 8 updated 6d ago

Torch-like deep learning framework for Javascript with support for tensors, autograd, optimizers, and other neural net constructs.

Creatify MCP 17 updated 10mo ago

Model Context Protocol server that exposes Creatify AI's video generation capabilities to AI assistants, enabling natural language video creation workflows.

Clustering.js

Clustering algorithms implemented in JavaScript for Node.js and the browser. [Deprecated]

Decision Trees 219 updated 24d ago

NodeJS Implementation of Decision Tree using ID3 Algorithm. [Deprecated]

DN2A 465 updated 2y ago

Digital Neural Networks Architecture. [Deprecated]

Gaussian Mixture Model 30 updated 1y ago

Unsupervised machine learning with multivariate Gaussian mixture model.

Node-fann 183 updated 9y ago

FANN (Fast Artificial Neural Network Library) bindings for Node.js [Deprecated]

Keras.js 5.0k updated 3y ago

Run Keras models in the browser, with GPU support provided by WebGL 2.

Kmeans.js 46 updated 12y ago

Simple JavaScript implementation of the k-means algorithm, for node.js and the browser. [Deprecated]

LDA.js 297 updated 1y ago

LDA topic modelling for Node.js

Learning.js

JavaScript implementation of logistic regression/c4.5 decision tree [Deprecated]

machinelearn.js 540 updated 1mo ago

Machine Learning library for the web, Node.js and developers

Node-SVM 301 updated 7y ago

Support Vector Machine for Node.js

Brain 8.0k (archived)

Neural networks in JavaScript [Deprecated]

Brain.js 14.9k updated 1y ago

Neural networks in JavaScript - continued community fork of Brain.

Bayesian-Bandit

Bayesian bandit implementation for Node and the browser. [Deprecated]

Synaptic 6.9k updated 5y ago

Architecture-free neural network library for Node.js and the browser.

kNear 48 updated 8y ago

JavaScript implementation of the k nearest neighbors algorithm for supervised learning.

NeuralN 274 updated 10y ago

C++ Neural Network library for Node.js. It has advantage on large dataset and multi-threaded training. [Deprecated]

kalman 114 updated 10y ago

Kalman filter for JavaScript. [Deprecated]

shaman 106 updated 10y ago

Node.js library with support for both simple and multiple linear regression. [Deprecated]

ml.js 2.7k updated 1y ago

Machine learning and numerical analysis tools for Node.js and the Browser!

ml5

Friendly machine learning for the web!

Pavlov.js 498 updated 8y ago

Reinforcement learning using Markov Decision Processes.

JSMLT 26 updated 3y ago

Machine learning toolkit with classification and clustering for Node.js; supports visualization (see visualml.io).

xgboost-node 48 updated 8y ago

Run XGBoost model and make predictions in Node.js.

Netron 32.6k updated 2d ago

Visualizer for machine learning models.

tensor-js

A deep learning library for the browser, accelerated by WebGL and WebAssembly.

WebDNN

Fast Deep Neural Network JavaScript Framework. WebDNN uses next generation JavaScript API, WebGPU for GPU execution, and WebAssembly for CPU execution.

Kandle 7 updated 1mo ago

A JavaScript Native PyTorch-aligned Machine Learning Framework, built from scratch on WebGPU.

Clusterfck 515 (archived)

Agglomerative hierarchical clustering implemented in JavaScript for Node.js and the browser. [Deprecated]

mil-tokyo

List of several machine learning libraries.

Misc Scripts / iPython Notebooks / Codebases

GreatCircle 77 updated 4y ago

Library for calculating great circle distance.

numpic 6.4k updated 1y ago
Allen Downey’s Think Bayes Code 1.7k updated 5y ago

Code repository for Think Bayes.

minidiff 1 updated 1mo ago

A slightly larger, somewhat feature-complete, PyTorch-inspired, NumPy implementation of a tensor reverse-mode automatic differentiation engine.

MiniGrad 97 updated 4y ago

A minimal, educational, Pythonic implementation of autograd (~100 loc).

Map/Reduce implementations of common ML algorithms

Jupyter notebooks that cover how to implement from scratch different ML algorithms (ordinary least squares, gradient descent, k-means, alternating least squares), using Python NumPy, and how to then make these implementations scalable using Map/Reduce and Spark.

CAEs for Data Assimilation 44 updated 5y ago

Convolutional autoencoders for 3D image/field compression applied to reduced order Data Assimilation.

handsonml 25.8k updated 7d ago

Fundamentals of machine learning in python.

SVM Explorer

Interactive SVM Explorer, using Dash and scikit-learn

pattern_classification
thinking stats 2 8 updated 11y ago
hyperopt 1.6k updated 11mo ago
2012-paper-diginorm 8 updated 11y ago
ipython-notebooks 575 updated 2d ago
data-science-ipython-notebooks

Continually updated Data Science Python Notebooks: Spark, Hadoop MapReduce, HDFS, AWS, Kaggle, scikit-learn, matplotlib, pandas, NumPy, SciPy, and various command lines.

decision-weights 33 updated 11y ago
Sarah Palin LDA 9 updated 14y ago

Topic Modelling the Sarah Palin emails.

Diffusion Segmentation

A collection of image segmentation algorithms based on diffusion methods.

Scipy Tutorials

SciPy tutorials. This is outdated, check out scipy-lecture-notes.

Crab 88 updated 14y ago

A recommendation engine library for Python.

BayesPy 109 updated 2y ago

Bayesian Inference Tools in Python.

scikit-learn tutorials 133 updated 14y ago

Series of notebooks for learning scikit-learn.

sentiment-analyzer 51 updated 14y ago

Tweets Sentiment Analyzer

sentiment_classifier 170 updated 4y ago

Sentiment classifier using word sense disambiguation.

group-lasso

Some experiments with the coordinate descent algorithm used in the (Sparse) Group Lasso model.

jProcessing 148 updated 5y ago

Kanji / Hiragana / Katakana to Romaji Converter. Edict Dictionary & parallel sentences Search. Sentence Similarity between two JP Sentences. Sentiment Analysis of Japanese Text. Run Cabocha(ISO--8859-1 configured) in Python.

mne-python-notebooks

IPython notebooks for EEG/MEG data processing using mne-python.

Neon Course 93 (archived)

IPython notebooks for a complete course around understanding Nervana's Neon.

pandas cookbook 7.0k updated 1y ago

Recipes for using Python's pandas library.

climin 183 updated 2mo ago

Optimization library focused on machine learning, pythonic implementations of gradient descent, LBFGS, rmsprop, adadelta and others.

Allen Downey’s Data Science Course

Code for Data Science at Olin College, Spring 2014.

Allen Downey’s Think Complexity Code 118 updated 1y ago

Code for Allen Downey's book Think Complexity.

Allen Downey’s Think OS Code 569 updated 5y ago

Text and supporting code for Think OS: A Brief Introduction to Operating Systems.

Dive into Machine Learning with Python Jupyter notebook and scikit-learn 11.4k (archived)

"I learned Python by hacking first, and getting serious later. I wanted to do this with Machine Learning. If this is your style, join me in getting a bit ahead of yourself."

TDB 1.4k updated 9y ago

TensorDebugger (TDB) is a visual debugger for deep learning. It features interactive, node-by-node debugging and visualization for TensorFlow.

Introduction to machine learning with scikit-learn 3.8k updated 2y ago

IPython notebooks from Data School's video tutorials on scikit-learn.

Introduction to Machine Learning with Python 8.1k updated 2y ago

Notebooks and code for the book "Introduction to Machine Learning with Python"

Pydata book 24.4k updated 5mo ago

Materials and IPython notebooks for "Python for Data Analysis" by Wes McKinney, published by O'Reilly Media

Homemade Machine Learning 24.4k updated 4mo ago

Python examples of popular machine learning algorithms with interactive Jupyter demos and math being explained

Prodmodel 58 updated 3y ago

Build tool for data science pipelines.

the-elements-of-statistical-learning 425 updated 1mo ago

This repository contains Jupyter notebooks implementing the algorithms found in the book and summary of the textbook.

Hyperparameter-Optimization-of-Machine-Learning-Algorithms 1.3k updated 3y ago

Code for hyperparameter tuning/optimization of machine learning and deep learning algorithms.

Heart_Disease-Prediction

Given clinical parameters about a patient, can we predict whether or not they have heart disease?

Flight Fare Prediction 1 updated 4y ago

This basically to gauge the understanding of Machine Learning Workflow and Regression technique in specific.

Keras Tuner

An easy-to-use, scalable hyperparameter optimization framework that solves the pain points of hyperparameter search.

BioPy 49 updated 2y ago

Biologically-Inspired and Machine Learning Algorithms in Python. [Deprecated]

Julia

General-Purpose Machine Learning

MachineLearning 118 updated 10y ago

Julia Machine Learning library. [Deprecated]

MLBase 186 updated 2y ago

A set of functions to support the development of machine learning algorithms.

PGM 53 updated 6y ago

A Julia framework for probabilistic graphical models.

DA 10 updated 5y ago

Julia package for Regularized Discriminant Analysis.

Regression 65 updated 8y ago

Algorithms for regression analysis (e.g. linear regression and logistic regression). [Deprecated]

Local Regression 110 updated 16d ago

Local regression, so smooooth!

Naive Bayes 8 updated 12y ago

Simple Naive Bayes implementation in Julia. [Deprecated]

Mixed Models 440 updated 16d ago

A Julia package for fitting (statistical) mixed-effects models.

Simple MCMC 12 (archived)

basic MCMC sampler implemented in Julia. [Deprecated]

Distances

Julia module for Distance evaluation.

Decision Tree 9 updated 3y ago

Decision Tree Classifier and Regressor.

Neural 48 updated 8y ago

A neural network in Julia.

MCMC 36 updated 13y ago

MCMC tools for Julia. [Deprecated]

Mamba 259 updated 4y ago

Markov chain Monte Carlo (MCMC) for Bayesian analysis in Julia.

GLM 633 updated 16d ago

Generalized linear models in Julia.

Gaussian Processes 319 updated 2mo ago

Julia package for Gaussian processes.

Online Learning 14 updated 11y ago

[Deprecated]

GLMNet 104 updated 2mo ago

Julia wrapper for fitting Lasso/ElasticNet GLM models using glmnet.

Clustering 373 updated 4mo ago

Basic functions for clustering data: k-means, dp-means, etc.

SVM 40 (archived)

SVM for Julia. [Deprecated]

Kernel Density 200 updated 16d ago

Kernel density estimators for Julia.

MultivariateStats

Methods for dimensionality reduction.

NMF 93 updated 1y ago

A Julia package for non-negative matrix factorization.

ANN

Julia artificial neural networks. [Deprecated]

Mocha 1.3k updated 7y ago

Deep Learning framework for Julia inspired by Caffe. [Deprecated]

XGBoost 304 updated 29d ago

eXtreme Gradient Boosting Package in Julia.

ManifoldLearning 94 updated 2y ago

A Julia package for manifold learning and nonlinear dimensionality reduction.

Merlin 146 updated 6y ago

Flexible Deep Learning Framework in Julia.

ROCAnalysis 33 updated 2y ago

Receiver Operating Characteristics and functions for evaluation probabilistic binary classifiers.

GaussianMixtures 107 updated 2mo ago

Large scale Gaussian Mixture Models.

ScikitLearn 559 updated 10mo ago

Julia implementation of the scikit-learn API.

Knet 1.4k updated 1y ago

Koç University Deep Learning Framework.

MLJ 1.9k updated 15d ago

A Julia machine learning framework.

CluGen 8 updated 1y ago

Multidimensional cluster generation in Julia.

Lua

General-Purpose Machine Learning

cephes 50 (archived)

Cephes mathematical functions library, wrapped for Torch. Provides and wraps the 180+ special mathematical functions from the Cephes mathematical library, developed by Stephen L. Moshier. It is used, among many other places, at the heart of SciPy. [Deprecated]

autograd 558 (archived)

Autograd automatically differentiates native Torch code. Inspired by the original Python version.

graph 38 updated 9y ago

Graph package for Torch. [Deprecated]

randomkit 34 updated 7y ago

Numpy's randomkit, wrapped for Torch. [Deprecated]

signal

A signal processing toolbox for Torch-7. FFT, DCT, Hilbert, cepstrums, stft.

nn 1.4k updated 5y ago

Neural Network package for Torch.

torchnet 992 (archived)

framework for torch which provides a set of abstractions aiming at encouraging code re-use as well as encouraging modular programming.

nngraph 302 updated 8y ago

This package provides graphical computation for nn library in Torch7.

nnx 98 updated 8y ago

A completely unstable and experimental package that extends Torch's builtin nn library.

rnn 943 updated 8y ago

A Recurrent Neural Network library that extends Torch's nn. RNNs, LSTMs, GRUs, BRNNs, BLSTMs, etc.

dpnn 193 updated 8y ago

Many useful features that aren't part of the main nn package.

dp 339 updated 9y ago

A deep learning library designed for streamlining research and development using the Torch7 distribution. It emphasizes flexibility through the elegant use of object-oriented design patterns. [Deprecated]

optim

An optimization library for Torch. SGD, Adagrad, Conjugate-Gradient, LBFGS, RProp and more.

unsup 86 updated 8y ago

A package for unsupervised learning in Torch. Provides modules that are compatible with nn (LinearPsd, ConvPsd, AutoEncoder, ...), and self-contained algorithms (k-means, PCA). [Deprecated]

manifold 142 updated 8y ago

A package to manipulate manifolds.

svm 44 updated 9y ago

Torch-SVM library. [Deprecated]

lbfgs 2 updated 13y ago

FFI Wrapper for liblbfgs. [Deprecated]

vowpalwabbit 2 updated 14y ago

An old vowpalwabbit interface to torch. [Deprecated]

OpenGM 8 updated 14y ago

OpenGM is a C++ library for graphical modelling, and inference. The Lua bindings provide a simple way of describing graphs, from Lua, and then optimizing them with OpenGM. [Deprecated]

spaghetti

Spaghetti (sparse linear) module for torch7 by @MichaelMathieu [Deprecated]

LuaSHKit 3 updated 12y ago

A Lua wrapper around the Locality sensitive hashing library SHKit [Deprecated]

kernel smoothing 5 updated 13y ago

KNN, kernel-weighted average, local linear regression smoothers. [Deprecated]

cutorch 339 updated 8y ago

Torch CUDA Implementation.

cunn 214 updated 6y ago

Torch CUDA Neural Network Implementation.

imgraph 22 updated 8y ago

An image/graph library for Torch. This package provides routines to construct graphs on images, segment them, build trees out of them, and convert them back to images. [Deprecated]

videograph 9 updated 12y ago

A video/graph library for Torch. This package provides routines to construct graphs on videos, segment them, build trees out of them, and convert them back to videos. [Deprecated]

saliency 7 updated 12y ago

code and tools around integral images. A library for finding interest points based on fast integral histograms. [Deprecated]

stitch 4 updated 14y ago

allows us to use hugin to stitch images and apply same stitching to a video sequence. [Deprecated]

sfm

A bundle adjustment/structure from motion package. [Deprecated]

fex 10 updated 12y ago

A package for feature extraction in Torch. Provides SIFT and dSIFT modules. [Deprecated]

OverFeat 601 updated 11y ago

A state-of-the-art generic dense feature extractor. [Deprecated]

wav2letter 6.4k updated 2mo ago

a simple and efficient end-to-end Automatic Speech Recognition (ASR) system from Facebook AI Research.

Lunum 40 updated 14y ago

[Deprecated]

Keras GPT Copilot 28 updated 2y ago

A python package that integrates an LLM copilot inside the keras model development workflow.

Objective C

General-Purpose Machine Learning

YCML 116 (archived)

A Machine Learning framework for Objective-C and Swift (OS X / iOS).

MLPNeuralNet 903 updated 9y ago

Fast multilayer perceptron neural network library for iOS and Mac OS X. MLPNeuralNet predicts new examples by trained neural networks. It is built on top of the Apple's Accelerate Framework, using vectorized operations and hardware acceleration if available. [Deprecated]

MAChineLearning 37 updated 6y ago

An Objective-C multilayer perceptron library, with full support for training through backpropagation. Implemented using vDSP and vecLib, it's 20 times faster than its Java equivalent. Includes sample code for use from Swift.

BPN-NeuralNetwork

It implemented 3 layers of neural networks ( Input Layer, Hidden Layer and Output Layer ) and it was named Back Propagation Neural Networks (BPN). This network can be used in products recommendation, user behavior analysis, data mining and data analysis. [Deprecated]

Multi-Perceptron-NeuralNetwork 24 updated 8y ago

It implemented multi-perceptrons neural network (ニューラルネットワーク) based on Back Propagation Neural Networks (BPN) and designed unlimited-hidden-layers.

KRHebbian-Algorithm 13 updated 10y ago

It is a non-supervisory and self-learning algorithm (adjust the weights) in the neural network of Machine Learning. [Deprecated]

KRKmeans-Algorithm 23 updated 9y ago

It implemented K-Means clustering and classification algorithm. It could be used in data mining and image compression. [Deprecated]

KRFuzzyCMeans-Algorithm 12 updated 10y ago

It implemented Fuzzy C-Means (FCM) the fuzzy clustering / classification algorithm on Machine Learning. It could be used in data mining and image compression. [Deprecated]

default

imbalanced-learn

Python module to perform under sampling and over sampling with various techniques.

Neural Networks and Deep Learning 17.5k updated 1y ago

Code samples for my book "Neural Networks and Deep Learning" [DEEP LEARNING].

Annoy

Approximate nearest neighbours implementation.

TPOT

Tool that automatically creates and optimizes machine learning pipelines using genetic programming. Consider it your personal data science assistant, automating a tedious part of machine learning.

pgmpy 3.2k updated 2d ago

A python library for working with Probabilistic Graphical Models.

DIGITS

The Deep Learning GPU Training System (DIGITS) is a web application for training deep learning models.

milk 603 (archived)

Machine learning toolkit focused on supervised classification. [Deprecated]

TFLearn

Deep learning library featuring a higher-level API for TensorFlow.

REP 700 updated 1y ago

an IPython-based environment for conducting data-driven research in a consistent and reproducible way. REP is not trying to substitute scikit-learn, but extends it and provides better user experience. [Deprecated]

rgf_python

Python bindings for Regularized Greedy Forest (Tree) Library.

skbayes

Python package for Bayesian Machine Learning with scikit-learn API.

fuku-ml

Simple machine learning library, including Perceptron, Regression, Support Vector Machine, Decision Tree and more, it's easy to use and easy to learn for beginners.

Xcessiv 1.3k updated 7y ago

A web-based application for quick, scalable, and automated hyperparameter tuning and stacked ensembling.

PyTorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch Lightning

The lightweight PyTorch wrapper for high-performance AI research.

PyTorch Lightning Bolts 1.8k (archived)

Toolbox of models, callbacks, and datasets for AI/ML researchers.

skorch

A scikit-learn compatible neural network library that wraps PyTorch.

Python Neural Networks

TResNet: Simple and powerful neural network library for python 167 updated 5y ago

Variety of supported types of Artificial Neural Network and learning algorithms.

Kinho

Simple API for Neural Network. Better for image processing with CPU/GPU + Transfer Learning.

nn_builder 165 updated 2y ago

nn_builder is a python package that lets you build neural networks in 1 line

NeuralTalk 5.5k updated 5y ago

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.

NeuralTalk 5.6k updated 8y ago

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences. [Deprecated]

Neuron 41 updated 2mo ago

Neuron is simple class for time series predictions. It's utilize LNU (Linear Neural Unit), QNU (Quadratic Neural Unit), RBF (Radial Basis Function), MLP (Multi Layer Perceptron), MLP-ELM (Multi Layer Perceptron - Extreme Learning Machine) neural networks learned with Gradient descent or LeLevenberg–Marquardt algorithm. [Deprecated]

Data Driven Code 30 updated 9y ago

Very simple implementation of neural networks for dummies in python without using any libraries, with detailed comments.

TResNet: High Performance GPU-Dedicated Architecture 478 updated 1y ago

TResNet models were designed and optimized to give the best speed-accuracy tradeoff out there on GPUs.

sequitur

PyTorch library for creating and training sequence autoencoders in just two lines of code

ANEE 1 updated 3mo ago

Adaptive Neural Execution Engine for transformers. Per-token sparse inference with dynamic layer skipping, profiler-based gating, and KV-cache-safe compute reduction.

Tools

More tools to improve the ML lifecycle: [Catalyst](https://github.com/catalyst-team/catalyst), [PachydermIO](https://www.pachyderm.io/). The following are GitHub-alike and targeting teams [Weights & Biases](https://www.wandb.com/), [Neptune.ai](https://neptune.ai/), [Comet.ml](https://www.comet.ml/), [Valohai.ai](https://valohai.com/), [DAGsHub](https://DAGsHub.com/).

Catalyst 3.4k updated 9mo ago

More tools to improve the ML lifecycle: Catalyst, PachydermIO. The following are GitHub-alike and targeting teams Weights & Biases, Neptune.ai, Comet.ml, Valohai.ai, DAGsHub.

m2cgen 3.0k updated 1y ago

A tool that allows the conversion of ML models into native code (Java, C, Python, Go, JavaScript, Visual Basic, C#, R, PowerShell, PHP, Dart) with zero dependencies.

CML 4.2k updated 9mo ago

A library for doing continuous integration with ML projects. Use GitHub Actions & GitLab CI to train and evaluate models in production like environments and automatically generate visual reports with metrics and graphs in pull/merge requests. Framework & language agnostic.

Chaos Genius

ML powered analytics engine for outlier/anomaly detection and root cause analysis.

MLEM

Version and deploy your ML models following GitOps principles

DockerDL 86 updated 16d ago

Ready to use deeplearning docker images.

Aqueduct 520 updated 2y ago

Aqueduct enables you to easily define, run, and manage AI & ML tasks on any cloud infrastructure.

Ambrosia 113 updated 2y ago

Ambrosia helps you clean up your LLM datasets using other LLMs.

promptfoo

Open-source LLM evaluation and red teaming framework. Test prompts, models, agents, and RAG pipelines. Run adversarial attacks (jailbreaks, prompt injection) and integrate security testing into CI/CD.

Agentic Radar 931 updated 3mo ago

Open-source CLI security scanner for agentic workflows. Scans your workflow’s source code, detects vulnerabilities, and generates an interactive visualization along with a detailed security report. Supports LangGraph, CrewAI, n8n, OpenAI Agents, and more.

ClawMoat 31 updated 8d ago

Open-source runtime security scanner for AI agents. Detects prompt injection, jailbreak, PII leakage, memory poisoning, and tool misuse. Zero deps, MIT licensed.

Agentic Signal 143 updated 15d ago

Visual AI agent workflow automation platform with local LLM integration. Build intelligent workflows using drag-and-drop, no cloud required.

Agentfield 1.1k updated yesterday

Open source Kubernetes-style control plane for deploying AI agents as distributed microservices, with built-in service discovery, durable workflows, and observability.

ScribePal 22 updated 4d ago

Chrome extension that uses local LLMs to assist with writing and drafting responses based on the context of your open tabs.

Local LLM NPC

Godot 4.x asset that enables NPCs to interact with players using local LLMs for structured, offline-first learning conversations in games.

Awesome Hugging Face Models 6 updated 19d ago

Curated list of top Hugging Face models for NLP, vision, and audio tasks with demos and benchmarks.

PraisonAI 5.7k updated yesterday

Production-ready Multi-AI Agents framework with self-reflection. Fastest agent instantiation (3.77μs), 100+ LLM support via LiteLLM, MCP integration, agentic workflows (route/parallel/loop/repeat), built-in memory, Python & JS SDKs.

RunAnywhere 10.4k updated 5d ago

Open-source SDK for running LLMs and multimodal models on-device across iOS, Android, and cross-platform apps.

HyperAgency 31 updated 21d ago

agentic AI operating system (h9y.ai) that replaces brittle/fragmented automations with long-lived, self-improving systems. Open-source, self-hosted/cloud, visual workflow, omni-channel, decentralized, extensible.

Bread Dataset Viewer

A VS Code extension for viewing and exploring large machine learning datasets (CSV, JSON, Parquet, etc.) directly within the editor without VS Code crashing in a clean UI.

Bread WandB Viewer

A VS Code extension to view Weights & Biases experiments, logs, and artifacts within the IDE, eliminating the need to switch to the web UI and keeping data private.

Misc

Infinity 4.5k updated yesterday

The AI-native database built for LLM applications, providing incredibly fast vector and full-text search. Developed using C++20

Qdrant 29.8k updated yesterday

Qdrant is open source vector similarity search engine with extended filtering support, written in Rust.

Localforge 353 updated 10mo ago

Is an open source on-prem AI coding autonomous assistant that lives inside your repo, edits and tests files at SSD speed. Think Claude Code but with UI. plug in any LLM (OpenAI, Gemini, Ollama, etc.) and let it work for you.

milvus 43.5k updated yesterday

Milvus is open source vector database for production AI, written in Go and C++, scalable and blazing fast for billions of embedding vectors.

Weaviate 15.9k updated yesterday

Weaviate is an open source vector search engine and vector database. Weaviate uses machine learning to vectorize and store data, and to find answers to natural language queries. With Weaviate you can also bring your custom ML models to production scale.

txtai 12.3k updated yesterday

Build semantic search applications and workflows.

DVC 15.5k updated 2d ago

Data Science Version Control is an open-source version control system for machine learning projects with pipelines support. It makes ML projects reproducible and shareable.

DVClive 189 updated 2d ago

Python library for experiment metrics logging into simply formatted local files.

VDP 2.3k updated 3d ago

open source visual data ETL to streamline the end-to-end visual data processing pipeline: extract unstructured visual data from pre-built data sources, transform it into analysable structured insights by Vision AI models imported from various ML platforms, and load the insights into warehouses or applications.

Kedro 10.8k updated yesterday

Kedro is a data and development workflow framework that implements best practices for data pipelines with an eye towards productionizing machine learning models.

Hamilton 2.4k updated 3d ago

a lightweight library to define data transformations as a directed-acyclic graph (DAG). It helps author reliable feature engineering and machine learning pipelines, and more.

Sacred 4.4k updated 5mo ago

Python tool to help you configure, organize, log and reproduce experiments. Like a notebook lab in the context of Chemistry/Biology. The community has built multiple add-ons leveraging the proposed standard.

Data Analysis / Data Visualization

DataComPy 639 updated 12d ago

A library to compare Pandas, Polars, and Spark data frames. It provides stats and lets users adjust for match accuracy.

AutoViz 1.9k updated 1y ago

AutoViz performs automatic visualization of any dataset with a single line of Python code. Give it any input file (CSV, txt or JSON) of any size and AutoViz will visualize it. See Medium article.

Mars 2.7k updated 2y ago

A tensor-based framework for large-scale data computation which is often regarded as a parallel and distributed version of NumPy.

Vaex 8.5k updated 25d ago

A high performance Python library for lazy Out-of-Core DataFrames (similar to Pandas), to visualize and explore big tabular datasets. Documentation can be found here.

PyMC 9.5k updated yesterday

Markov Chain Monte Carlo sampling toolkit.

zipline

A Pythonic algorithmic trading library.

SymPy 14.5k updated 2d ago

A Python library for symbolic mathematics.

statsmodels 11.3k updated 6d ago

Statistical modelling and econometrics in Python.

bokeh 20.4k updated yesterday

Interactive Web Plotting for Python.

altair 10.3k updated yesterday

A Python to Vega translator.

PyQtGraph 4.3k updated 16d ago

A pure-python graphics and GUI library built on PyQt4 / PySide and NumPy.

Blaze

NumPy and Pandas interface to Big Data.

emcee

The Python ensemble sampling toolkit for affine-invariant MCMC.

vispy

GPU-based high-performance interactive OpenGL 2D/3D data visualization library.

SparklingPandas

Pandas on PySpark (POPS).

bqplot

An API for plotting in Jupyter (IPython).

Superset 71.1k updated yesterday

A data exploration platform designed to be visual, intuitive, and interactive.

Dora 649 updated 7mo ago

Tools for exploratory data analysis in Python.

SOMPY 552 updated 3y ago

Self Organizing Map written in Python (Uses neural networks for data analysis).

somoclu 277 updated 3mo ago

Massively parallel self-organizing maps: accelerate training on multicore CPUs, GPUs, and clusters, has python API.

HDBScan 102 updated 8y ago

implementation of the hdbscan algorithm in Python - used for clustering

scikit-plot 2.4k updated 1y ago

A visualization library for quick and easy generation of common plots in data analysis and machine learning.

Bowtie 768 updated 6y ago

A dashboard library for interactive visualizations using flask socketio and react.

lime 12.1k updated 1y ago

Lime is about explaining what machine learning classifiers (or models) are doing. It is able to explain any black box classifier, with two or more classes.

PyCM

PyCM is a multi-class confusion matrix library written in Python that supports both input data vectors and direct matrix, and a proper tool for post-classification model evaluation that supports most classes and overall statistics parameters

Dash 24.5k updated yesterday

A framework for creating analytical web applications built on top of Plotly.js, React, and Flask

Lambdo 25 updated 5y ago

A workflow engine for solving machine learning problems by combining in one analysis pipeline (i) feature engineering and machine learning (ii) model training and prediction (iii) table population and column evaluation via user-defined (Python) functions.

TensorWatch 3.5k updated 8d ago

Debugging and visualization tool for machine learning and data science. It extensively leverages Jupyter Notebook to show real-time visualizations of data in running processes such as machine learning training.

dowel 35 updated 2y ago

A little logger for machine learning research. Output any object to the terminal, CSV, TensorBoard, text logs on disk, and more with just one call to logger.log().

Flama 288 updated 1mo ago

Ignite your models into blazing-fast machine learning APIs with a modern framework.

DataVisualization 49 updated 3y ago

A GitHub Repository Where you can Learn Datavisualizatoin Basics to Intermediate level.

ParaMonte

A general-purpose Python library for Bayesian data analysis and visualization via serial/parallel Monte Carlo and MCMC simulations. Documentation can be found here.

Open Mining 1.3k (archived)

Business Intelligence (BI) in Python (Pandas web interface) [Deprecated]

d3py

A plotting library for Python, based on D3.js.

PyDexter

Simple plotting for Python. Wrapper for D3xterjs; easily render charts in-browser.

ggplot

Same API as ggplot2 for R. [Deprecated]

ggfortify 538 updated 5mo ago

Unified interface to ggplot2 popular R packages.

Kartograph.py 1.0k updated 5y ago

Rendering beautiful SVG maps in Python.

pycascading 221 (archived)

[Deprecated]

Petrel 246 updated 3y ago

Tools for writing, submitting, debugging, and monitoring Storm topologies in pure Python.

windML 129 updated 2y ago

A Python Framework for Wind Energy Analysis and Prediction.

cerebro2

A web-based visualization and debugging platform for NuPIC. [Deprecated]

NuPIC Studio 96 updated 5y ago

An all-in-one NuPIC Hierarchical Temporal Memory visualization and debugging super-tool! [Deprecated]

ipychart 131 updated 1y ago

The power of Chart.js in Jupyter Notebook.

pastalog 421 updated 9y ago

Simple, realtime visualization of neural network training performance.

visualize_ML 205 updated 9y ago

A python package for data exploration and data analysis. [Deprecated]

Kaggle Competition Source Code

open-solution-home-credit

source code and experiments results for Home Credit Default Risk.

open-solution-googleai-object-detection 46 (archived)

source code and experiments results for Google AI Open Images - Object Detection Track.

open-solution-salt-identification 121 (archived)

source code and experiments results for TGS Salt Identification Challenge.

open-solution-ship-detection

source code and experiments results for Airbus Ship Detection Challenge.

open-solution-data-science-bowl-2018 155 (archived)

source code and experiments results for 2018 Data Science Bowl.

open-solution-value-prediction

source code and experiments results for Santander Value Prediction Challenge.

open-solution-toxic-comments 156 (archived)

source code for Toxic Comment Classification Challenge.

wiki challenge

An implementation of Dell Zhang's solution to Wikipedia's Participation Challenge on Kaggle.

kaggle insults

Kaggle Submission for "Detecting Insults in Social Commentary".

kaggle_acquire-valued-shoppers-challenge 66 updated 12y ago

Code for the Kaggle acquire valued shoppers challenge.

kaggle-cifar 44 updated 11y ago

Code for the CIFAR-10 competition at Kaggle, uses cuda-convnet.

kaggle-blackbox 116 updated 11y ago

Deep learning made easy.

kaggle-accelerometer 15 updated 11y ago

Code for Accelerometer Biometric Competition at Kaggle.

kaggle-advertised-salaries 55 updated 11y ago

Predicting job salaries from ads - a Kaggle competition.

kaggle amazon 25 updated 11y ago

Amazon access control challenge.

kaggle-bestbuy_big 8 updated 11y ago

Code for the Best Buy competition at Kaggle.

kaggle-bestbuy_small 6 updated 11y ago
Kaggle Dogs vs. Cats 65 updated 5y ago

Code for Kaggle Dogs vs. Cats competition.

Kaggle Galaxy Challenge 497 updated 11y ago

Winning solution for the Galaxy Challenge on Kaggle.

Kaggle Gender 22 updated 11y ago

A Kaggle competition: discriminate gender based on handwriting.

Kaggle Merck 10 updated 11y ago

Merck challenge at Kaggle.

Kaggle Stackoverflow 44 updated 8y ago

Predicting closed questions on Stack Overflow.

wine-quality 26 updated 10y ago

Predicting wine quality.

Reinforcement Learning

DeepMind Lab 7.3k updated 3y ago

DeepMind Lab is a 3D learning environment based on id Software's Quake III Arena via ioquake3 and other open source software. Its primary purpose is to act as a testbed for research in artificial intelligence, especially deep reinforcement learning.

Gymnasium 11.6k updated 2d ago

A library for developing and comparing reinforcement learning algorithms (successor of gym).

Serpent.AI 7.0k (archived)

Serpent.AI is a game agent framework that allows you to turn any video game you own into a sandbox to develop AI and machine learning experiments. For both researchers and hobbyists.

ViZDoom 2.0k updated 22d ago

ViZDoom allows developing AI bots that play Doom using only the visual information (the screen buffer). It is primarily intended for research in machine visual learning, and deep reinforcement learning, in particular.

Roboschool

Open-source software for robot simulation, integrated with OpenAI Gym.

Retro

Retro Games in Gym

SLM Lab 1.3k updated yesterday

Modular Deep Reinforcement Learning framework in PyTorch.

Coach

Reinforcement Learning Coach by Intel AI Lab enables easy experimentation with state of the art Reinforcement Learning algorithms

garage 2.1k updated 2y ago

A toolkit for reproducible reinforcement learning research

metaworld 1.8k updated 2mo ago

An open source robotics benchmark for meta- and multi-task reinforcement learning

Maze 287 updated yesterday

Application-oriented deep reinforcement learning framework addressing real-world decision problems.

RLlib 41.8k updated yesterday

RLlib is an industry level, highly scalable RL library for tf and torch, based on Ray. It's used by companies like Amazon and Microsoft to solve real-world decision making problems at scale.

DI-engine

DI-engine is a generalized Decision Intelligence engine. It supports most basic deep reinforcement learning (DRL) algorithms, such as DQN, PPO, SAC, and domain-specific algorithms like QMIX in multi-agent RL, GAIL in inverse RL, and RND in exploration problems.

Gym4ReaL 48 updated 8mo ago

Gym4ReaL is a comprehensive suite of realistic environments designed to support the development and evaluation of RL algorithms that can operate in real-world scenarios. The suite includes a diverse set of tasks exposing RL algorithms to a variety of practical challenges.

Ruby

Rust

Swift

General-Purpose Machine Learning

Bender 1.8k updated 2y ago

Fast Neural Networks framework built on top of Metal. Supports TensorFlow models.

Swift AI 6.1k updated 9y ago

Highly optimized artificial intelligence and machine learning library written in Swift.

Swift for Tensorflow

a next-generation platform for machine learning, incorporating the latest research across machine learning, compilers, differentiable programming, systems design, and beyond.

BrainCore 379 updated 9y ago

The iOS and OS X neural network framework.

swix 590 (archived)

A bare bones library that includes a general matrix language and wraps some OpenCV for iOS development. [Deprecated]

AIToolbox 802 updated 5y ago

A toolbox framework of AI modules written in Swift: Graphs/Trees, Linear Regression, Support Vector Machines, Neural Networks, PCA, KMeans, Genetic Algorithms, MDP, Mixture of Gaussians.

MLKit

A simple Machine Learning Framework written in Swift. Currently features Simple Linear Regression, Polynomial Regression, and Ridge Regression.

Swift Brain 339 updated 8y ago

The first neural network / machine learning library written in Swift. This is a project for AI algorithms in Swift for iOS and OS X development. This project includes algorithms focused on Bayes theorem, neural networks, SVMs, Matrices, etc...

Perfect TensorFlow 167 updated 5y ago

Swift Language Bindings of TensorFlow. Using native TensorFlow models on both macOS / Linux.

PredictionBuilder

A library for machine learning that builds predictions using a linear regression.

Awesome CoreML 586 updated 6y ago

A curated list of pretrained CoreML models.

Awesome Core ML Models 7.0k updated 9mo ago

A curated list of machine learning models in CoreML format.