Research Python Packages

rouge-score

Google Research

yacs

YACS -- Yet Another Configuration System

sb3-contrib

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code

rtdl-num-embeddings

(NeurIPS 2022) On Embeddings for Numerical Features in Tabular Deep Learning

pymbar

Python implementation of the multistate Bennett acceptance ratio (MBAR)

scann

Google Research

arxiv-mcp-server

A local MCP server for agent literature work. Original-LaTeX section reads, BibTeX from arXiv metadata, and topic watches. Papers stay on disk. Search is optional.

pypartmc

Python (and C++) interface to PartMC with Jupyter/Python, Julia and Matlab examples

gpt-researcher

An autonomous agent that conducts deep research on any data using any LLM providers

tableone

Create "Table 1" for research papers in Python

scitex-io

Universal scientific data I/O with a plugin format registry

optax-shampoo

Google Research

granular

Fast dataset format and loader

khoj

Your AI second brain. Self-hostable. Get answers from the web or your docs. Build custom agents, schedule automations, do deep research. Turn any online or local LLM into your personal, autonomous AI (gpt, claude, gemini, llama, qwen, mistral). Get started - free.

openpmd-api

:floppy_disk: C++ & Python API for Scientific I/O

wntr

An EPANET compatible python package to simulate and analyze water distribution networks under disaster scenarios.

scinumtools3

Define, validate, and report scientific parameters with units and provenance across C++, Python, CLI, CMake, and REST.

dcm2niix

dcm2nii DICOM to NIfTI converter: compiled versions available from NITRC

papers-dl

A command line application for downloading scientific papers.

pymeasure

Scientific measurement library for instruments, experiments, and live-plotting

carla

Open-source simulator for autonomous driving research.

pii-codex

A research python package for detecting, categorizing, and assessing the severity of personal identifiable information (PII)

cadence-net

A brain that learns from live experience. System 1 by default: perception, memory, plasticity and action. Optional System 2 for self-observation and slow corrections.

dallinger

Laboratory automation for the behavioral and social sciences; human culture on a chip.