spectral_galerkin_heat.backends.cupy_backend ============================================ .. py:module:: spectral_galerkin_heat.backends.cupy_backend .. autoapi-nested-parse:: GPU math backend: CuPy arrays with Numba CUDA kernels. This module imports :mod:`cupy`, so it is imported lazily (only from :func:`spectral_galerkin_heat.backends.get_backend`) to keep CuPy an optional dependency. Classes ------- .. autoapisummary:: spectral_galerkin_heat.backends.cupy_backend.CupyBackend Module Contents --------------- .. py:class:: CupyBackend Bases: :py:obj:`spectral_galerkin_heat.backends.base.MathBackend` GPU backend: CuPy arrays and Numba CUDA kernels.