Multi Node Multi GPU Offline Inference and metric calculation library
Install this library using pip
:
pip install crossfit
git clone https://github.com/rapidsai/crossfit.git
cd crossfit
pip install --extra-index-url https://pypi.nvidia.com ".[cuda12x]"
Usage instructions go here.
To contribute to this library, first create a conda environment with the necessary dependencies:
cd crossfit
mamba env create -f conda/environments/cuda_dev.yaml
conda activate crossfit_dev
Now install the dependencies and test dependencies:
pip install -e '.[test]'
To run the tests:
pytest