Price of Freedom: Exploring Tradeoffs between Expressivity and Computational Efficiency in Equivariant Tensor Products

Paper: https://openreview.net/forum?id=0HHidbjwcf

Installation

pip install -r requirements.txt

Tetris Experiments

bash shell/run_tetris_experiments.sh

Benchmarking

CPU timing

bash shell/run_cpu_timing.sh

GPU timing

bash shell/run_gpu_timing.sh

Nsight Compute (GPU-only)

Nsight Compute installation needed. Make sure the GPU counters are enabled

bash shell/run_nsight_profiling.sh