rusty1s / pytorch_scatter

Related projects ⓘ

Alternatives and complementary repositories for pytorch_scatter

• rusty1s / pytorch_sparse
  PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
  ☆1,011Updated last week
• rusty1s / pytorch_cluster
  PyTorch Extension Library of Optimized Graph Cluster Algorithms
  ☆827Updated last week
• lightaime / deep_gcns_torch
  Pytorch Repo for DeepGCNs (ICCV'2019 Oral, TPAMI'2021), DeeperGCN (arXiv'2020) and GNN1000(ICML'2021): https://www.deepgcns.org
  ☆1,133Updated 2 years ago
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,056Updated this week
• juho-lee / set_transformer
  Pytorch implementation of set transformer
  ☆546Updated 4 years ago
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,644Updated last year
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,021Updated 3 months ago
• snap-stanford / ogb
  Benchmark datasets, data loaders, and evaluators for graph machine learning
  ☆1,945Updated 9 months ago
• facebookresearch / higher
  higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…
  ☆1,594Updated 2 years ago
• Lightning-AI / torchmetrics
  Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,141Updated this week
• Oldpan / Pytorch-Memory-Utils
  pytorch memory track code
  ☆1,003Updated 3 years ago
• jeanfeydy / geomloss
  Geometric loss functions between point clouds, images and volumes
  ☆598Updated 10 months ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆774Updated 10 months ago
• vahidk / tfrecord
  Standalone TFRecord reader/writer with PyTorch data loaders
  ☆867Updated 3 months ago
• facebookincubator / submitit
  Python 3.8+ toolbox for submitting jobs to Slurm
  ☆1,303Updated 2 months ago
• lucidrains / performer-pytorch
  An implementation of Performer, a linear attention-based transformer, in Pytorch
  ☆1,098Updated 2 years ago
• Tiiiger / SGC
  official implementation for the paper "Simplifying Graph Convolutional Networks"
  ☆832Updated 2 years ago
• HobbitLong / PyContrast
  PyTorch implementation of Contrastive Learning methods
  ☆1,943Updated last year
• mit-han-lab / torchsparse
  [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
  ☆1,225Updated last week
• graphdeeplearning / graphtransformer
  Graph Transformer Architecture. Source code for "A Generalization of Transformer Networks to Graphs", DLG-AAAI'21.
  ☆892Updated 3 years ago
• facebookresearch / fvcore
  Collection of common code that's shared among different research projects in FAIR computer vision team.
  ☆2,031Updated this week
• NVIDIA / MinkowskiEngine
  Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
  ☆2,493Updated 8 months ago
• geoopt / geoopt
  Riemannian Adaptive Optimization Methods with pytorch optim
  ☆853Updated 6 months ago
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆727Updated 2 years ago
• rbgirshick / yacs
  YACS -- Yet Another Configuration System
  ☆1,281Updated 2 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆700Updated 6 months ago
• graphdeeplearning / benchmarking-gnns
  Repository for benchmarking graph neural networks
  ☆2,527Updated last year
• QUVA-Lab / e2cnn
  E(2)-Equivariant CNNs Library for Pytorch
  ☆604Updated last year
• williamleif / graphsage-simple
  Simple reference implementation of GraphSAGE.
  ☆998Updated 4 years ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,395Updated this week