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.

chronos-forecasting

Chronos: Pretrained Models for Time Series Forecasting

neptune-query

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

tabpfn

⚡ TabPFN: Foundation Model for Tabular Data ⚡

skforecast

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

nixtla

TimeGPT-2.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.

tabpfn-client

⚡ Easy API access to the tabular foundation model TabPFN ⚡

osam

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

terratorch

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

colossalai

Making large AI models cheaper, faster and more accessible

torchxrayvision

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

mrnavax

Practical Python tools for the AI-leverage layers in mRNA cancer therapy. Stdlib-only: codon optimization, neoantigen screening, TrialGPT-style trial matching, and LNP composition advice.

microsoft-aurora

Implementation of the Aurora model for Earth system forecasting

tfc-t0

An open-weights time-series forecasting foundation model from The Forecasting Company.

tirex-ts

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

meds-reader

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

timecopilot-chronos-forecasting

Chronos: Pretrained Models for Time Series Forecasting

autodistill

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

timecopilot-tirex

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

visbench

Probe any vision backbone across high-, mid- and low-level computer vision tasks

diffbio

End-to-end differentiable bioinformatics for JAX/Flax NNX: soft Smith-Waterman alignment, read mapping and assembly, variant calling, RNA-seq, single-cell, epigenomics, CRISPR, metabolomics, multi-omics, protein and RNA structure, molecular dynamics and drug-discovery operators composed into trainable pipelines on the Avitai stack.