Flux
Flux: Elegant machine learning with Julia - Published in JOSS (2018)
PyCM
PyCM: Multiclass confusion matrix library in Python - Published in JOSS (2018)
samgeo
samgeo: A Python package for segmenting geospatial data with the Segment Anything Model (SAM) - Published in JOSS (2023)
TorchGAN
TorchGAN: A Flexible Framework for GAN Training and Evaluation - Published in JOSS (2021)
pytorch-widedeep
pytorch-widedeep: A flexible package for multimodal deep learning - Published in JOSS (2023)
Physics-Informed Neural networks for Advanced modeling
Physics-Informed Neural networks for Advanced modeling - Published in JOSS (2023)
DeepRiver
DeepRiver: A Deep Learning Library for Data Streams - Published in JOSS (2025)
DeepRank2
DeepRank2: Mining 3D Protein Structures with Geometric Deep Learning - Published in JOSS (2024)
InvertibleNetworks.jl
InvertibleNetworks.jl: A Julia package for scalable normalizing flows - Published in JOSS (2024)
SelfEEG
SelfEEG: A Python library for Self-Supervised Learning in Electroencephalography - Published in JOSS (2024)
Delve
Delve: Neural Network Feature Variance Analysis - Published in JOSS (2022)
FastVPINNs
FastVPINNs: An efficient tensor-based Python library for solving partial differential equations using hp-Variational Physics Informed Neural Networks - Published in JOSS (2024)
MapReader
MapReader: Open software for the visual analysis of maps - Published in JOSS (2024)
FTorch
FTorch: a library for coupling PyTorch models to Fortran - Published in JOSS (2025)
TorchMetrics - Measuring Reproducibility in PyTorch
TorchMetrics - Measuring Reproducibility in PyTorch - Published in JOSS (2022)
GNS
GNS: A generalizable Graph Neural Network-based simulator for particulate and fluid modeling - Published in JOSS (2023)
Kinetic.jl
Kinetic.jl: A portable finite volume toolbox for scientific and neural computing - Published in JOSS (2021)
ChainoPy
ChainoPy: A Python Library for Discrete Time Markov Chain Based Stochastic Analysis - Published in JOSS (2024)
DIRECT
DIRECT: Deep Image REConstruction Toolkit - Published in JOSS (2022)
ATHENA
ATHENA: A Fortran package for neural networks - Published in JOSS (2024)
deepface
A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python
transformers
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
ViMag
ViMag: A Visual Vibration Analysis Toolbox - Published in JOSS (2023)
sahi
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots
hivemind
Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.
boxmot
BoxMOT: Pluggable SOTA multi-object tracking modules modules for segmentation, object detection and pose estimation models
deepmd-kit
A deep learning package for many-body potential energy representation and molecular dynamics
pypots
A Python toolkit/library for reality-centric machine/deep learning and data mining on partially-observed time series, including SOTA neural network models for scientific analysis tasks of imputation/classification/clustering/forecasting/anomaly detection/cleaning on incomplete industrial (irregularly-sampled) multivariate TS with NaN missing values
datasets
🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools
pykeen
🤖 A Python library for learning and evaluating knowledge graph embeddings
BayesFlow
BayesFlow: Amortized Bayesian Workflows With Neural Networks - Published in JOSS (2023)
edsnlp
Modular, fast NLP framework, compatible with Pytorch and spaCy, offering tailored support for French clinical notes.
flaml
A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.
quantus
Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations
decimer
DECIMER Image Transformer is a deep-learning-based tool designed for automated recognition of chemical structure images. Leveraging transformer architectures, the model converts chemical images into SMILES strings, enabling the digitization of chemical data from scanned documents, literature, and patents.
autopytorch
Automatic architecture search and hyperparameter optimization for PyTorch
tts
🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production
DEEPaaS API
DEEPaaS API: a REST API for Machine Learning and Deep Learning models - Published in JOSS (2019)
deepxde
A library for scientific machine learning and physics-informed learning
proglearn
NeuroData's package for exploring and using progressive learning algorithms
onnx2tf
Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf). I don't need a Star, but give me a pull request.
nrel-sup3r
The Super-Resolution for Renewable Resource Data (sup3r) software uses generative adversarial networks to create synthetic high-resolution wind and solar spatiotemporal data from coarse low-resolution inputs.
stepcount
Improved Step Counting via Foundation Models for Wrist-Worn Accelerometers
pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
cvnn
Library to help implement a complex-valued neural network (cvnn) using tensorflow as back-end
awesome-dataset-distillation
A curated list of awesome papers on dataset distillation and related applications.
litgpt
20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.
lumin
LUMIN - a deep learning and data science ecosystem for high-energy physics.
awesome-implicit-nerf-robotics
A comprehensive list of Implicit Representations and NeRF papers relating to Robotics/RL domain, including papers, codes, and related websites
yaib
🧪Yet Another ICU Benchmark: a holistic framework for the standardization of clinical prediction model experiments. Provide custom datasets, cohorts, prediction tasks, endpoints, preprocessing, and models. Paper: https://arxiv.org/abs/2306.05109
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
analogvnn
A fully modular framework for modeling and optimizing analog neural networks
CounterfactualExplanations
A package for Counterfactual Explanations and Algorithmic Recourse in Julia.
symbac
Accurate segmentation of bacterial microscope images using deep learning synthetically generated image data.
torchxrayvision
TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.
mmaction2
OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
smart-transformers
Smart Transformers are a versatile machine learning tool that can be integrated with Pytorch, TensorFlow, and JAX. Smart transformers provide accurate computations required for cryptographic algorithms. These transformers is that they are independent modules, making it efficient to experiment with various research projects related to cryptanalysis
ionmob
An open-source prediction framework for peptide ion collision cross section (CCS) values with python.
active-learning-as-a-service
A scalable & efficient active learning/data selection system for everyone.
avalanche-lib
Avalanche: an End-to-End Library for Continual Learning based on PyTorch.
monailabel
MONAI Label is an intelligent open source image labeling and learning tool.
gt4sd
GT4SD, an open-source library to accelerate hypothesis generation in the scientific discovery process.
deepflash2
A deep-learning pipeline for segmentation of ambiguous microscopic images.
habitat
🔮 Execution time predictions for deep neural network training iterations across different GPUs.
paddlescience
PaddleScience is SDK and library for developing AI-driven scientific computing applications based on PaddlePaddle.
gnina-torch
🔥 PyTorch implementation of GNINA scoring function for molecular docking
deepke
[EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and Construction
ncnn
ncnn is a high-performance neural network inference framework optimized for the mobile platform
fsrs4anki-optimizer
A modern Anki custom scheduling based on Free Spaced Repetition Scheduler algorithm
torchopt
TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
mtg-jamendo-dataset
Metadata, scripts and baselines for the MTG-Jamendo dataset