GitHub projects

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

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

The lightweight PyTorch wrapper for high-performance AI research.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

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.

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

Python bindings for Regularized Greedy Forest (Tree) Library.

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]

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

Machine learning toolkit focused on supervised classification. [Deprecated]

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

A python library for working with Probabilistic Graphical Models.

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.

Approximate nearest neighbours implementation.

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

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

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

Deep Learning In Python. [Deprecated]

Evolutionary algorithm framework.

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

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

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

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

Python Environment for Bayesian Learning. [Deprecated]

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]

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

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.

Open source platform for deploying machine learning models in production.

Theano based library for deep and recurrent neural networks.

Genetic algorithm framework. [Deprecated]

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.

Machine learning for NeuroImaging in Python.

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

Bolt Online Learning Toolbox. [Deprecated]

Restricted Boltzmann Machines in Python. [DEEP LEARNING]

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

Book on Bayesian Analysis.

A Python library for implementing a Recommender System.

A flexible, fast recommender engine. [Deprecated]

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

Another Python Machine Learning Library.

Topic modelling toolkit. [Deprecated]

Interactive topic model visualization/interpretation framework.

Topic Modelling for Humans.

A scikit-learn native toolkit for nonprofit and academic-medical-center fundraising analytics: leakage-safe donor propensity, lapse, planned-giving, wealth-screening and revenue-forecasting estimators.

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

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.