Repository Index

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

• autodiff
• deep-learning
• machine-learning
• rust

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

• java
• gpu
• deeplearning
• neural-nets

Open3D: A Modern Library for 3D Data Processing

• mesh-processing
• computer-graphics
• opengl
• cpp

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

AI Toolkit for Healthcare Imaging

• healthcare-imaging
• deep-learning
• medical-image-computing
• medical-image-processing

A PyTorch Library for Accelerating 3D Deep Learning Research

• pytorch
• 3d-deep-learning
• neural-networks
• differentiable-rendering

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

• quantum
• machine-learning
• deep-learning
• neural-network

Accelerated deep learning R&D

• deep-learning
• reinforcement-learning
• machine-learning
• computer-vision

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

• gpu
• numerical
• elixir
• tensor

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

• machine-learning
• python
• statistics
• scikit-learn

Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).

• python
• genetic-algorithm
• optimization
• numpy

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)

• deep-learning
• pytorch
• deep-chemistry
• graph-neural-network

OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research

• computational-chemistry
• computational-biology
• deep-learning
• deep-neural-networks

Engine for AI/ML/Data tracking, visualization, explainability, drift detection, and dashboards for Polyaxon.

• pandas
• pandas-summary
• dataframes
• data-science

Tools for computational pathology

• machine-learning
• digital-pathology
• computational-pathology
• biomedical-image-processing

This is a Pytorch implementation of the paper: Self-Supervised Graph Transformer on Large-Scale Molecular Data

PyTorch Connectomics: segmentation toolbox for EM connectomics

• computer-vision
• neuroscience
• connectomics
• pytorch

MCP server for spatial transcriptomics analysis through natural language interfaces.

• bioinformatics
• cell-communication
• deconvolution
• machine-learning

Multi-stage Riemannian flow matching for physically valid molecular docking, with GNINA scoring, PoseBusters filtering, CLI inference, and benchmarks.

• alphafold3
• docking
• drug-design
• drug-discovery