Machine Learning

IBM watsonx.ai API Client

A Python library for probabilistic modeling and inference

A suite of visual analysis and diagnostic tools for machine learning.

TensorFlow is an open source machine learning framework for everyone.

Generic API for dispatch to Pyro backends.

Reversible Data Transforms

QUick and DIrty Domain Adaptation

Generalist model for NER (Extract any entity types from texts)

Supercharging Machine Learning

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

A machine learning toolkit dedicated to time-series data

Image augmentation library for deep neural networks

Fit interpretable models. Explain blackbox machine learning.

Set of libraries for ML training loops in JAX.

Contains core functionality for Azure Machine Learning pipelines, which are configurable machine learning workflows.

tensorflow/datasets is a library of datasets ready to use with TensorFlow.

A Python library for audio data augmentation. Inspired by albumentations. Useful for machine learning.

Collection of training and inference decision forest algorithms.

ClearML - Auto-Magical Experiment Manager, Version Control, and MLOps for AI

PyFlux: A time-series analysis library for Python

Scalable machine learning based time series forecasting

libtpu is the core library that enables machine learning frameworks like JAX, PyTorch, and TensorFlow to execute models on Google Cloud TPUs. It provides core functionality for compilation, inter-chip communication (ICI), and runtime execution, and includes a set of SDK primitives for direct TPU interaction and deployment. Learn more about Cloud TPUs at https://cloud.google.com/tpu.

✂️ Fast slice finding for Machine Learning model debugging.

Gin-Config: A lightweight configuration library for Python

Continuous control environments and MuJoCo Python bindings.

Used for automatically finding the best machine learning model and its parameters.

Kedro-Viz helps visualise Kedro data and analytics pipelines

Fit interpretable models. Explain blackbox machine learning.

Pyro PPL on NumPy

Integration tools for running scikit-learn on Spark

Converts Machine Learning models to ONNX

NeMo - a toolkit for Conversational AI

Fiddle: A Python-first configuration library

Probabilistic Programming in Python: Bayesian Modeling and Probabilistic Machine Learning with Theano

Used to build, optimize, and manage their machine learning workflows.

Used to build and run machine learning workflows upon the Azure Machine Learning service.

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.

Azure Machine Learning inferencing server.

Is a metapackage that is used internally by Azure Machine Learning

TensorFlow Hub is a library to foster the publication, discovery, and consumption of reusable parts of machine learning models.

Removed: please install "tensorflow" instead.

simple library to manage a dataset of shards to train machine learning models

A set of tools, related to machine learning in production.

H2O, Fast Scalable Machine Learning, for python

A Python interface for Reinforcement Learning environments.

Scalable Nearest Neighbor search library

A web scraping library based on LangChain which uses LLM and direct graph logic to create scraping pipelines.

Local Interpretable Model-Agnostic Explanations for machine learning classifiers

This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/Transformers.

This package is intended to provide a uniform schema for common machine learning applications, as well as a set of decorators that can be used to aid in web based ML prediction applications.