yzh119

BPT

Source code of paper "BP-Transformer: Modelling Long-Range Context via Binary Partitioning"

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

FeatGraph

ffi-benchmark

Benchmark different ffi solutions

amted

basic_verilog

Must-have verilog systemverilog modules

calyx

Intermediate Language (IL) for Hardware Accelerator Generators

compiler-and-arch

A list of tutorials, paper, talks, and open-source projects for emerging compiler and architecture

cutlass

CUDA Templates for Linear Algebra Subroutines

dlpack

RFC for common in-memory tensor structure and operator interface for deep learning system

dslabs

Distributed Systems Labs and Framework

egg

egg is a flexible, high-performance e-graph library

hemem-defrag

Defragmentation during migragation in HeMem (CSE599f Sp 2022)

long-range-arena

Long Range Arena for Benchmarking Efficient Transformers

nestedtensor

[EXPERIMENTAL] Tools for the concurrent manipulation of variably sized Tensors.

nn_pruning

Prune a model while finetuning or training.

ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning

open-gpu-kernel-modules

NVIDIA Linux open GPU kernel module source

pizza

qlib

Qlib is an AI-oriented quantitative investment platform, which aims to realize the potential, empower the research, and create the value of AI technologies in quantitative investment. With Qlib, you can easily try your ideas to create better Quant investment strategies.

Remmy

A simple but usable RPC framework

reticle

RoseTTAFold

This package contains deep learning models and related scripts for RoseTTAFold

synr

A library for syntactically rewriting Python programs, pronounced (sinner).

tenset

TLCBench

Benchmark scripts for TVM

tlcpack-sphinx-addon

Additional sphinx theming on top of rtd theme with support for header and footer

torchsparse

[MLSys'22] TorchSparse: Efficient Point Cloud Inference Engine

toyrpc

tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation