# Copyright 2024-2026 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
DISTUTILS_USE_PEP517=setuptools
DISTUTILS_SINGLE_IMPL=1
PYTHON_COMPAT=( python3_{12..14} )
inherit distutils-r1 pypi
DESCRIPTION="Stochastic differential equation solvers with GPU support"
HOMEPAGE="https://github.com/google-research/torchsde https://pypi.org/project/torchsde/"
LICENSE="Apache-2.0"
SLOT="0"
KEYWORDS="~amd64"
RDEPEND="
	|| (
		sci-ml/pytorch[${PYTHON_SINGLE_USEDEP}]
		sci-ml/pytorch-rocm[${PYTHON_SINGLE_USEDEP}]
		sci-ml/pytorch-cuda[${PYTHON_SINGLE_USEDEP}]
	)
	$(python_gen_cond_dep 'dev-python/numpy[${PYTHON_USEDEP}]')
"