FTorch
FTorch: a library for coupling PyTorch models to Fortran - Published in JOSS (2025)
VisualTorch
VisualTorch: Streamlining Visualization for PyTorch Neural Network Architectures - Published in JOSS (2024)
pykeen
🤖 A Python library for learning and evaluating knowledge graph embeddings
torchrl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
https://github.com/metatensor/metatrain
Training and evaluating machine learning models for atomistic systems.
https://github.com/awslabs/gluonts
Probabilistic time series modeling in Python
benchmarl
BenchMARL is a library for benchmarking Multi-Agent Reinforcement Learning (MARL). BenchMARL allows to quickly compare different MARL algorithms, tasks, and models while being systematically grounded in its two core tenets: reproducibility and standardization.
spyrit
A Python toolbox for deep image reconstruction, with emphasis on single-pixel imaging.
eurocropsml
EuroCropsML is a ready-to-use benchmark dataset for few-shot crop type classification using Sentinel-2 imagery.
https://github.com/althonos/torch-treecrf
A PyTorch implementation of Tree-structured Conditional Random Fields.
https://github.com/astrazeneca/chemicalx
A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)
evolutionary-diffusion
Applying Evolutionary Computing to Embeddings of Diffusion Models
https://github.com/superduper-io/superduper
Superduper: End-to-end framework for building custom AI applications and agents.
orca
ORCA: Oceanic Recognition & Classification Application for sea-life analysis systems.
https://github.com/zama-ai/concrete-ml
Concrete ML: Privacy Preserving ML framework using Fully Homomorphic Encryption (FHE), built on top of Concrete, with bindings to traditional ML frameworks.
https://github.com/cpu-ds/unike
基于 OpenKE-PyTorch 开发的知识图谱嵌入工具包,支持跨平台运行,具备自动超参数搜索、高效并行训练以及实验结果记录功能,为研究与应用提供强大助力。
https://github.com/chrislisbon/torchcnnbuilder
Framework for the automatic creation of CNN architectures
videomae
[NeurIPS'22] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training
https://github.com/complight/spectrack
❇️ SpecTrack Enhances Precision in Multi-Rotation Tracking
https://github.com/arya-gaj/your-phone-can-spot-fashion-v1
A lightweight yet powerful system that analyzes short-form videos in real time to identify fashion products by combining computer vision and natural language processing, all processed locally.
https://github.com/complight/holobeam_multiholo
:goggles: HoloBeam: Paper-Thin Near-Eye Displays
echotorch
A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.
mdi-llm
Implementation of Model-Distributed Inference for Large Language Models, built on top of LitGPT
videoswin
Keras 3 Implementation of Video Swin Transformers for 3D Video Modeling
mukh
A comprehensive face analysis library that provides unified APIs for various face-related tasks
MRI-NUFFT
MRI-NUFFT: Doing non-Cartesian MRI has never been easier - Published in JOSS (2025)
https://github.com/braun-steven/spn-pytorch-experiments
Experiments on the integration of Sum-Product Networks into Deep Neural Networks.
https://github.com/cosmoquester/torch-lightning-template
This repository is template for my pytorch lightning project.
gempy
GemPy is an open-source, Python-based 3-D structural geological modeling software, which allows the implicit (i.e. automatic) creation of complex geological models from interface and orientation data. It also offers support for stochastic modeling to address parameter and model uncertainties.
3d-viz-score-cam
Visualizing 3D ResNet for Medical Image Classification With Score-CAM
https://github.com/ipc-lab/kaira
A PyTorch-based toolkit for simulating communication systems
https://github.com/anapnoe/stable-diffusion-webui-ux
Stable Diffusion web UI UX
https://github.com/augustunderground/ganarchist
Gallium Nitrid Half Bridge Switching Loss Prediction
https://github.com/augustunderground/edelwace
Reinforcement Learning Agents for Analog Circuit Sizing in Haskell.