aiamplitudes-common-public 0.1.dev51

A library of common utils for the AI Amplitudes project

aiamplitudes_common_public

Common public utils library for the "AI for Scattering Amplitudes" project. The package should be a simple install with pip, but please notify me if this is not the case! If there are issues with hatch, simply install an editable version (pip install -e .) into your local conda environment.

src/ init.py- import-level objects and functions. These include the Phi2 and Phi3 symbols, the front and back spaces and relations in various weights and formats, run-polynomials, and their coeffs. For example use-cases, please explore the Amplitudes_101 jupyter notebook!

download_data.py - contains the scripts to download a (compressed) version of the data files. Call download_all(DATA_DIR_NAME) to download the files. This is useful for EDA and generators, but should not be called every time- for large-scale generation, it may be useful to pre-install these files in a Docker container.

file_readers.py - functions to read the files. convert() is a multipurpose function that is called often, while the others are more specialized to different cases. Most of these are called within functions and should not need to be called by the typical user.

fbspaces.py - Utils for the front and back spaces. This includes coproducts, lookups, etc.

rels_utils.py- Table of homogeneous relations (as defined in the "Bootstrapping a Stress Tensor..." paper), relation eval functions, etc.- this is called often!

commonclasses.py - Common wrappers used for generating and processing relation-instances.

polynomial_utils.py - functions related to analysis of the "d-ray" polynomials mentioned in the "Recurrent Features of Amplitudes..." paper.

notebook/ Amplitudes_101.ipynb- a notebook that displays some simple exploratory data analysis functions of this package.