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

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

Installation

We recommend using uv:

uv sync
source .venv/bin/activate
uv pip install -e .

Alternatively, you can install the dependencies manually:

python -m venv .venv
source .venv/bin/activate
pip install -r requirements.in

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