uwsampl/pytorch uwsampl/pytorch

  • Stars
    star
    6
  • Rank 2,539,965 (Top 51 %)
  • Language
    C++
  • License
    Other
  • Created about 5 years ago
  • Updated about 1 year ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
uwsampl/pytorch
github

uwsampl

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Dynamic tensor rematerialization, implemented as a pytorch fork

More Repositories

1

SparseTIR

SparseTIR: Sparse Tensor Compiler for Deep Learning
Python
131
star
2

dtr-prototype

Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
Python
129
star
3

nexus

C++
81
star
4

relay-aot

An experimental ahead of time compiler for Relay.
Python
51
star
5

tutorial

A self-contained version of the tutorial which can be easily cloned and viewed by others.
Jupyter Notebook
25
star
6

sparsetir-artifact

Repository for artifact evaluation of ASPLOS 2023 paper "SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning"
Python
23
star
7

lastlayer

Towards Hardware and Software Continuous Integration
Rust
13
star
8

relay-bench

A repository containing examples and benchmarks for Relay.
Python
7
star
9

ml4mlsys

Course website for UW CSE 599M: ML for ML Systems in Spring 2023
HTML
7
star
10

dtr

Dynamic Tensor Rematerialization
6
star
11

uwsampl.github.io

The UW SAMPL group's website.
HTML
4
star
12

vta-distro

A repository to cache VTA bistreams
4
star
13

3la-ir-example

Example showing the different stages of Relay and TVM IR for a single program.
Python
2
star
14

web-data

Website-data for SAML-related projects
2
star
15

egraphs-ai-experiments

Rust
2
star
16

vta-posit

Scala
1
star
17

3la-tvm

Fork of TVM for 3LA BYOC extensions
Python
1
star