Foundation Models Python Packages

toons

A high-performance TOON (Token Oriented Object Notation) parser and serializer for Python.

pyod

A Python library for anomaly detection across tabular, time series, graph, text, image, and audio data. 60+ detectors, benchmark-backed ADEngine orchestration, and an agentic workflow for AI agents.

neptune-fetcher

Neptune Fetcher is designed to separate data retrieval capabilities from the regular neptune package. This separation makes data fetching more efficient and improves performance.

neptune-query

Read-only API for fetching metadata tracked with the Neptune logging client.

chronos-forecasting

Chronos: Pretrained Models for Time Series Forecasting

tabpfn

⚡ TabPFN: Foundation Model for Tabular Data ⚡

tabicl

TabICLv2: A state-of-the-art tabular foundation model

nixtla

TimeGPT-1: production ready pre-trained Time Series Foundation Model for forecasting and anomaly detection. Generative pretrained transformer for time series trained on over 100B data points. It's capable of accurately predicting various domains such as retail, electricity, finance, and IoT with just a few lines of code 🚀.

mattersim

MatterSim: A deep learning atomistic model across elements, temperatures and pressures.

skforecast

Python library for time series forecasting using scikit-learn compatible models, statistical methods, and foundation models

torchxrayvision

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

osam

Get up and running with SAM1,2,3, EfficientSAM, YOLO-World, and other promptable vision models locally.

tabpfn-client

⚡ Easy API access to the tabular foundation model TabPFN ⚡

terratorch

A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).

colossalai

Making large AI models cheaper, faster and more accessible

tirex-ts

TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning

microsoft-aurora

Implementation of the Aurora model for Earth system forecasting

autodistill

Images to inference with no labeling (use foundation models to train supervised models).

meds-reader

A high-performance database and Python reader interface for MEDS datasets

liquid-audio

Liquid Audio - Speech-to-Speech audio models by Liquid AI

pfnstudio

Open-source toolkit + marketplace for prior-fitted foundation models (PFNs)

helical

A framework for state-of-the-art pre-trained bio foundation models on genomics and transcriptomics modalities.

py-data-juicer

Data processing for and with foundation models! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷

gigaam

Foundational Model for Speech Recognition Tasks