# NeoMLP

## Setup environment

To run the experiments, first create a clean virtual environment and install the requirements.

```bash
mamba create -n neomlp python=3.10
mamba activate neomlp
mamba install pytorch==2.2.1 torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
mamba install pyg==2.5.0 pytorch-scatter -c pyg
mamba install h5py
```

Install the repo:

```bash
cd neomlp
pip install -e .
cd experiments/visualization_utils && python setup.py build_ext --inplace
```
