Open Source Tools for AI-Projects

Displaying 1 - 10 of 34

DeepSpeed by microsoft


DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

nni by microsoft


An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

fairlearn by fairlearn


A Python package to assess and improve fairness of machine learning models.

foolbox by bethgelab


A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX



Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams

netron by lutzroeder


Visualizer for neural network, deep learning, and machine learning models

interpret by interpretml


Fit interpretable models. Explain blackbox machine learning.

pycaret by pycaret


An open-source, low-code machine learning library in Python

plotly.py by plotly


The interactive graphing library for Python (includes Plotly Express) :sparkles:

shap by slundberg


A game theoretic approach to explain the output of any machine learning model.