Index

A

abs (in module lab.generic) abstract() (in module lab.util) add (in module lab.generic) all (in module lab.generic) any (in module lab.generic)

argsort (in module lab.generic) as_tf (in module lab.tensorflow.custom) as_torch (in module lab.torch.custom) as_tuple (in module lab.util) autograd_register() (in module lab.autograd.custom)

B

batch_computation() (in module lab.util) bvn_cdf (in module lab.generic)

bvn_cdf() (in module lab.bvn_cdf) (in module lab.custom)

C

cast (in module lab.generic) choice (in module lab.random) chol (in module lab.linear_algebra) cholesky (in module lab.linear_algebra)

cholesky_solve (in module lab.linear_algebra) cholsolve (in module lab.linear_algebra) concat (in module lab.shaping) concat2d (in module lab.shaping) cos (in module lab.generic)

D

default_dtype (in module lab.types) det (in module lab.linear_algebra) diag (in module lab.shaping)

divide (in module lab.generic) dot (in module lab.linear_algebra) dtype (in module lab.types)

E

epsilon (in module lab.linear_algebra) erf (in module lab.generic) ew_dists (in module lab.linear_algebra) ew_dists2 (in module lab.linear_algebra) ew_sums (in module lab.linear_algebra)

ew_sums2 (in module lab.linear_algebra) exp (in module lab.generic) expand_dims (in module lab.shaping) expm (in module lab.linear_algebra) expm() (in module lab.custom) eye (in module lab.generic)

F

flatten (in module lab.shaping)

G

ge (in module lab.generic)

gt (in module lab.generic)

I

identity (in module lab.generic) inv (in module lab.linear_algebra)

isnan (in module lab.generic) isscalar (in module lab.shaping) issubdtype (in module lab.types)

J

jax_register() (in module lab.jax.custom)

K

kron (in module lab.linear_algebra)

L

lab module lab.autograd module lab.autograd.custom module lab.autograd.generic module lab.autograd.linear_algebra module lab.autograd.random module lab.autograd.shaping module lab.bvn_cdf module lab.custom module lab.generic module lab.jax module lab.jax.custom module lab.jax.generic module lab.jax.linear_algebra module lab.jax.random module lab.jax.shaping module lab.linear_algebra module lab.numpy module lab.numpy.generic module lab.numpy.linear_algebra module lab.numpy.random module lab.numpy.shaping module

lab.random module lab.shaping module lab.tensorflow module lab.tensorflow.custom module lab.tensorflow.generic module lab.tensorflow.linear_algebra module lab.tensorflow.random module lab.tensorflow.shaping module lab.torch module lab.torch.custom module lab.torch.generic module lab.torch.linear_algebra module lab.torch.random module lab.torch.shaping module lab.types module lab.util module le (in module lab.generic) leaky_relu (in module lab.generic) length (in module lab.shaping) linspace (in module lab.generic) log (in module lab.generic) log_2_pi (in module lab.generic) logdet (in module lab.linear_algebra) logm (in module lab.linear_algebra) logm() (in module lab.custom) logsumexp (in module lab.generic) lt (in module lab.generic)

M

matmul (in module lab.linear_algebra) max (in module lab.generic) maximum (in module lab.generic) mean (in module lab.generic) min (in module lab.generic) minimum (in module lab.generic) mm (in module lab.linear_algebra) module lab lab.autograd lab.autograd.custom lab.autograd.generic lab.autograd.linear_algebra lab.autograd.random lab.autograd.shaping lab.bvn_cdf lab.custom lab.generic lab.jax lab.jax.custom lab.jax.generic lab.jax.linear_algebra lab.jax.random lab.jax.shaping lab.linear_algebra lab.numpy lab.numpy.generic lab.numpy.linear_algebra lab.numpy.random lab.numpy.shaping lab.random lab.shaping lab.tensorflow lab.tensorflow.custom lab.tensorflow.generic lab.tensorflow.linear_algebra lab.tensorflow.random lab.tensorflow.shaping lab.torch lab.torch.custom lab.torch.generic lab.torch.linear_algebra lab.torch.random lab.torch.shaping lab.types lab.util

multiply (in module lab.generic)

N

nan (in module lab.generic)

negative (in module lab.generic)

O

one (in module lab.generic)

ones (in module lab.generic) outer (in module lab.linear_algebra)

P

pi (in module lab.generic) power (in module lab.generic) pw_dists (in module lab.linear_algebra)

pw_dists2 (in module lab.linear_algebra) pw_sums (in module lab.linear_algebra) pw_sums2 (in module lab.linear_algebra)

R

rand (in module lab.random) randn (in module lab.random) range (in module lab.generic)

rank (in module lab.shaping) reg (in module lab.linear_algebra) relu (in module lab.generic) reshape (in module lab.shaping)

S

s_bvn_cdf() (in module lab.bvn_cdf) (in module lab.custom) s_expm() (in module lab.custom) s_logm() (in module lab.custom) s_toeplitz_solve() (in module lab.custom) scan (in module lab.generic) set_random_seed (in module lab.random) shape (in module lab.shaping) sigmoid (in module lab.generic) sign (in module lab.generic) sin (in module lab.generic)

size (in module lab.shaping) softplus (in module lab.generic) solve (in module lab.linear_algebra) sort (in module lab.generic) sqrt (in module lab.generic) squeeze (in module lab.shaping) stack (in module lab.shaping) std (in module lab.generic) subtract (in module lab.generic) sum (in module lab.generic) svd (in module lab.linear_algebra)

T

T (in module lab.linear_algebra) t (in module lab.linear_algebra) take (in module lab.shaping) tan (in module lab.generic) tanh (in module lab.generic) tensorflow_register() (in module lab.tensorflow.custom) tile (in module lab.shaping) to_numpy (in module lab.generic)

toeplitz_solve (in module lab.linear_algebra) toeplitz_solve() (in module lab.custom) toepsolve (in module lab.linear_algebra) torch_register() (in module lab.torch.custom) trace (in module lab.linear_algebra) transpose (in module lab.linear_algebra) triangular_solve (in module lab.linear_algebra) tril_to_vec (in module lab.shaping) trisolve (in module lab.linear_algebra)

U

unstack (in module lab.shaping)

uprank (in module lab.shaping)

V

vec_to_tril (in module lab.shaping)

Z

zero (in module lab.generic)

zeros (in module lab.generic)