memory-efficient topic

List memory-efficient repositories

Dragon

176
Stars
52
Forks
Watchers

Dragon: A Computation Graph Virtual Machine Based Deep Learning Framework.

memory-efficient-attention-pytorch

342
Stars
32
Forks
Watchers

Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"

sgscript

63
Stars
7
Forks
Watchers

SGScript Scripting Engine

GC-DPR

127
Stars
20
Forks
Watchers

Train Dense Passage Retriever (DPR) with a single GPU

optimized_transducer

68
Stars
11
Forks
Watchers

Memory efficient transducer loss computation

madoka-python

25
Stars
2
Forks
Watchers

Memory-efficient Count-Min Sketch Counter (based on Madoka C++ library)

Social-Implicit

82
Stars
8
Forks
Watchers

Code for: "Social-Implicit: Rethinking Trajectory Prediction Evaluation and The Effectiveness of Implicit Maximum Likelihood Estimation" Accepted @ ECCV2022

scarf

100
Stars
12
Forks
Watchers

Toolkit for highly memory efficient analysis of single-cell RNA-Seq, scATAC-Seq and CITE-Seq data. Analyze atlas scale datasets with millions of cells on laptop.

ringattention

593
Stars
43
Forks
Watchers

Transformers with Arbitrarily Large Context

CAME

65
Stars
5
Forks
Watchers

The official implementation of "CAME: Confidence-guided Adaptive Memory Optimization"