LeMA: Multi-GPU Levenberg-Marquardt Optimizer Powered by NVIDIA cuPyNumeric and Legate

Get Started

Install dependencies

# cuPyNumeric, Legate
CONDA_OVERRIDE_CUDA="12.2" \
  conda install -c conda-forge -c legate cupynumeric

Build LeMA

  pip install -e .