requirements.txt

## Code quality
pytest==7.0.1
black==22.3.0

## Standard science stack
h5py
hdf5plugin
matplotlib
numpy>=1.25.0,<=1.26.4 # numpy>=1.25.0 is needed for pytorch-optimizer, numpy<=1.26.4 is needed for numba
scipy

## Standard utils
tqdm
wget

## Deep learning
einops
opt_einsum
torch==2.2.0
torch_geometric==2.4.0
torch-ema
git+https://github.com/kozistr/pytorch_optimizer # most recent pytorch_optimizer contains ADOPT; replace this with simply 'pytorch_optimizer>=3.2.0' after the next release

## Experiment management
hydra-core
mlflow==2.4.1 # avoid database version issues
optuna

## Project-specific: general
clifford==1.4.0
numba==0.58.1 # otherwise has conflicts
xformers==0.0.24 # there exists known compatibility issues with xformers on Mac systems
torchdiffeq # for CFM sampling

#weaver packages
tqdm>=4.41.0
PyYAML>=5.4
awkward0>=0.15.5
uproot>=4.2.0,<5.2.0
awkward>=1.8.0
lz4
xxhash