Index

_

__call__() (waxmorph.jax.gnn.GNS method) (waxmorph.jax.gnn.GraphNetworkBlock method) (waxmorph.jax.mlp.MLP method)

B

backward() (waxmorph.torch.warp_autograd.WarpDiffusionStep static method) (waxmorph.torch.warp_autograd.WarpMechStep static method) build_edge_features() (in module waxmorph.jax.graph) (in module waxmorph.torch.graph) build_edge_index() (in module waxmorph.jax.graph) (in module waxmorph.torch.graph)

build_graph() (in module waxmorph.jax.graph) (in module waxmorph.torch.graph) build_node_features() (in module waxmorph.jax.graph) (in module waxmorph.torch.graph)

C

chamfer_distance() (in module waxmorph.jax.losses) (in module waxmorph.torch.losses)

chem_step() (in module waxmorph.simulator) count_neighbors_step() (in module waxmorph.simulator)

D

diffusion_step_differentiable() (in module waxmorph.emulator)

draw_3d_view() (waxmorph.render.MPLInterface static method) (waxmorph.render.PyVistaInterface static method)

F

forward() (waxmorph.torch.gnn.GNS method) (waxmorph.torch.gnn.GraphNetworkBlock method) (waxmorph.torch.mlp.MLP method) (waxmorph.torch.warp_autograd.WarpDiffusionStep static method) (waxmorph.torch.warp_autograd.WarpMechStep static method)

G

gen_key_array() (in module waxmorph.simulator) GNS (class in waxmorph.jax.gnn) (class in waxmorph.torch.gnn)

GraphNetworkBlock (class in waxmorph.jax.gnn) (class in waxmorph.torch.gnn) growth_step() (in module waxmorph.simulator)

L

load() (waxmorph.jax.gnn.GNS class method) (waxmorph.torch.gnn.GNS class method)

load_mesh() (in module waxmorph.data)

M

make_samples_loss() (in module waxmorph.torch.losses) make_sinkhorn_loss() (in module waxmorph.jax.losses) mech_step_sticky() (in module waxmorph.simulator) mech_step_sticky_differentiable() (in module waxmorph.emulator) mech_step_sticky_implicit() (in module waxmorph.simulator) MLP (class in waxmorph.jax.mlp) (class in waxmorph.torch.mlp) module waxmorph waxmorph.constants waxmorph.data waxmorph.emulator waxmorph.gnn waxmorph.graph waxmorph.jax waxmorph.jax.gnn waxmorph.jax.graph waxmorph.jax.losses waxmorph.jax.mlp waxmorph.jax.train waxmorph.jax.warp_autograd waxmorph.losses waxmorph.mlp waxmorph.render waxmorph.simulator waxmorph.torch waxmorph.torch.gnn waxmorph.torch.graph waxmorph.torch.losses waxmorph.torch.mlp waxmorph.torch.train waxmorph.torch.warp_autograd waxmorph.train

MPLInterface (class in waxmorph.render)

N

normalize_mesh() (in module waxmorph.data)

P

PyVistaInterface (class in waxmorph.render)

R

RenderInterface (class in waxmorph.render)

S

sample_mesh_pair() (in module waxmorph.data) sample_mesh_sequence() (in module waxmorph.data) sample_volume() (in module waxmorph.data)

save() (waxmorph.jax.gnn.GNS method) (waxmorph.torch.gnn.GNS method) squared_loss() (in module waxmorph.jax.losses) (in module waxmorph.torch.losses)

T

train() (in module waxmorph.jax.train) (in module waxmorph.torch.train) (in module waxmorph.train) TrainConfig (class in waxmorph.jax.train) (class in waxmorph.torch.train) (class in waxmorph.train)

TrainResult (class in waxmorph.jax.train) (class in waxmorph.torch.train) (class in waxmorph.train)

W

warp_diffusion_step() (in module waxmorph.jax.warp_autograd) warp_mech_step() (in module waxmorph.jax.warp_autograd) WarpDiffusionStep (class in waxmorph.torch.warp_autograd) WarpMechStep (class in waxmorph.torch.warp_autograd) WarpMovieRenderer (class in waxmorph.render) waxmorph module waxmorph.constants module waxmorph.data module waxmorph.emulator module waxmorph.gnn module waxmorph.graph module waxmorph.jax module waxmorph.jax.gnn module waxmorph.jax.graph module waxmorph.jax.losses module waxmorph.jax.mlp module waxmorph.jax.train module

waxmorph.jax.warp_autograd module waxmorph.losses module waxmorph.mlp module waxmorph.render module waxmorph.simulator module waxmorph.torch module waxmorph.torch.gnn module waxmorph.torch.graph module waxmorph.torch.losses module waxmorph.torch.mlp module waxmorph.torch.train module waxmorph.torch.warp_autograd module waxmorph.train module write_frame_from_numpy() (waxmorph.render.WarpMovieRenderer method) write_frame_from_state() (waxmorph.render.WarpMovieRenderer method)