Name		Name	Last commit message	Last commit date
.github/workflows		.github/workflows
LICENSES		LICENSES
scripts		scripts
src/tensorforge		src/tensorforge
tests		tests
tools		tools
.gitignore		.gitignore
.markdownlint.yaml		.markdownlint.yaml
.pre-commit-config.yaml		.pre-commit-config.yaml
.pylintrc		.pylintrc
AUTHORS		AUTHORS
LICENSE		LICENSE
README.md		README.md
REUSE.toml		REUSE.toml
header.py		header.py
pyproject.toml		pyproject.toml
requirements.txt		requirements.txt
setup.cfg		setup.cfg

Repository files navigation

TensorForge

A code generator for small (batched) tensor contraction kernels on GPUs.

Installation

TensorForge is a Python package, i.e. you may run

pip install .

to install it.

About

A kernel generator for small, batched tensor contractions and more

Custom properties

2 stars

7 watching

Report repository

Releases

Packages

Contributors

Languages