Repository Index

PrismQuant: Optimal Null-Space Rotations for Grouped Quantizers

• int4
• large-language-models
• pytorch
• quantization

Graph to sequence implemented in Pytorch combining Graph convolutional networks and opennmt-py

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

• gnn
• graph-neural-netowrks
• signed-networks
• directed-networks

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

• adversarial-attacks
• graph-convolutional-networks
• graph-neural-networks
• pytorch

RootPainter: Deep Learning Segmentation of Biological Images with Corrective Annotation

• biological-images
• root
• painter
• segmentation

PyTorch implementation of the MIDAS algorithm for single-cell multimodal data integration (Nature Biotechnology 2024).

• deep-generative-model
• deep-learning
• single-cell-data-integration
• single-cell-multiomics

An Easy To Use PyTorch Computer Vision Library

• computer-vision
• deep-learning
• pytorch
• pytorch-lightning

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

This repository contains the official PyTorch implementation of GeoDA algorithm. GeoDA is a Black-box attack to generate adversarial example for image classifiers.

YOLOv5 - most advanced vision AI model for object detection. Natively implemented in PyTorch and exportable to TFLite for use in edge solutions.

PyTorch implementation of "TGSA: Protein-Protein Association-Based Twin Graph Neural Networks for Drug Response Prediction with Similarity Augmentation"

Tool for suggesting lemma names in Coq verification projects

• coq
• machine-learning
• name-generation
• pytorch

Panoptic segmentation algorithms for 2D and 3D electron microscopy images

• deep-learning
• semantic-segmentation
• segmentation
• pytorch

A minimalistic codebase for training NLG models from HuggingFace Transformers using PyTorch Lightning.

• bart
• generation
• gpt-2
• language-model