Multi Armed Bandit Python Packages

baytune

A simple, extensible library for developing AutoML systems

pybandits

Python library for Multi-Armed Bandits

mabalgs

:bust_in_silhouette: Multi-Armed Bandit Algorithms Library (MAB) :cop:

mab-algorithm

Code template for multi-armed bandit algorithm

smpybandits

🔬 Research Framework for Single and Multi-Players 🎰 Multi-Arms Bandits (MAB) Algorithms, implementing all the state-of-the-art algorithms for single-player (UCB, KL-UCB, Thompson...) and multi-player (MusicalChair, MEGA, rhoRand, MCTop/RandTopM etc).. Available on PyPI: https://pypi.org/project/SMPyBandits/ and documentation on

kvfleet

KV-cache-aware intelligent routing for self-hosted and hybrid LLM fleets. Route requests using model quality, latency, cost, policy, and live GPU state.

ab

A/B Testing Framework for Python (with optional Multi-armed bandit implementation)

slots

A multi-armed bandit library for Python

improveai

Contextual Multi-Armed Bandit Platform for Scoring, Ranking & Decisions

thompson

Thompson is Python package to evaluate the multi-armed bandit problem. In addition to thompson, Upper Confidence Bound (UCB) algorithm, and randomized results are also implemented.

contextual-bandits-algos

A comprehensive Python library implementing a variety of contextual and non-contextual multi-armed bandit algorithms—including LinUCB, Epsilon-Greedy, Upper Confidence Bound (UCB), Thompson Sampling, KernelUCB, NeuralLinearBandit, and DecisionTreeBandit—designed for reinforcement learning applications