BG Trees on the GPU

Welcome to the documentation of BG Trees!

BG Trees is a flexible and modular library that makes it easier to compute scattering amplitudes using the Berends-Giele recursion algorithm. As higher-order calculations get more complex and resource-intensive, new strategies need to be developed. Being able to compute amplitudes with various types of arithmetic and in an arbitrary number of dimension can be proven crucial for loop calculations.

By tapping into the power of GPUs, BG Trees helps cut down on the normally high costs of these calculations, making it much more feasible to tackle advanced problems.

Installation

The code, library and CUDA kernels can be installed with.

pip install git+https://github.com/Amps-GPU/BG-Trees

This command will automatically install the tensorflow[and-cuda] package and it will attempt to compile the CUDA kernels if the nvcc compiler is available.

Quick start

Indices and tables

Modules Documentation