Source code for gensbi.diffusion.solver.solver

from abc import ABC, abstractmethod

from jax import Array




class Solver(ABC):
    """Abstract base class for solvers."""



    @abstractmethod
    def sample(self, key, x_1: Array) -> Array:
        ...