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.