KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
Primary LanguagePythonMIT LicenseMIT