Index

A

arccos() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) arccosh() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) arcsin() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

arcsinh() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) arctan() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) arctanh() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

C

cat() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) cos() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

cosh() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

E

elbo() (in module surkit.losses.losses_pytorch) exp() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

F

forward() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

G

get() (in module surkit.data.sampling) (in module surkit.losses.losses_pytorch) grad() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

grad_h() (surkit.gradient.sobelfilter.SobelFilter method) grad_v() (surkit.gradient.sobelfilter.SobelFilter method)

I

is_number() (in module surkit.data.dataset) is_tensor() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

L

load() (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) load_dataset() (in module surkit.data.dataset) log() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

log10() (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) log2() (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

M

module surkit.backend surkit.backend.backend surkit.backend.jax_bkd surkit.backend.pytorch_bkd surkit.conditions surkit.data surkit.data.dataset surkit.data.sampling surkit.gradient surkit.gradient.sobelfilter surkit.losses.losses_pytorch

N

np_to_tensor() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

P

power() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

R

random_sampler() (in module surkit.data.sampling)

S

save() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) sin() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) sinh() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) SobelFilter (class in surkit.gradient.sobelfilter) sqrt() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) squeeze() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd) surkit.backend module surkit.backend.backend module

surkit.backend.jax_bkd module surkit.backend.pytorch_bkd module surkit.conditions module surkit.data module surkit.data.dataset module surkit.data.sampling module surkit.gradient module surkit.gradient.sobelfilter module surkit.losses.losses_pytorch module

T

tan() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

tanh() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

U

uniform_sampler() (in module surkit.data.sampling) unsqueeze() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

W

weight_1d() (in module surkit.data.sampling)

weight_2d() (in module surkit.data.sampling)

Z

zeros() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)

zeros_like() (in module surkit.backend.backend) (in module surkit.backend.jax_bkd) (in module surkit.backend.pytorch_bkd)