Index

A

adjust_schedule() (in module mdot_tnt.mdot)

B

BatchedTruncatedNewtonProjector (class in mdot_tnt.batched)

blocked_rounded_cost() (in module mdot_tnt.lowmem) blocked_transport_cost() (in module mdot_tnt.lowmem)

C

chi_sinkhorn() (mdot_tnt.lowmem.LowMemoryTruncatedNewtonProjector method) (mdot_tnt.truncated_newton.TruncatedNewtonProjector method)

E

euclidean() (in module mdot_tnt.lowmem)

L

LowMemoryTruncatedNewtonProjector (class in mdot_tnt.lowmem)

M

mdot() (in module mdot_tnt.mdot) mdot_lowmem() (in module mdot_tnt.lowmem) mdot_tnt.batched module mdot_tnt.lowmem module mdot_tnt.mdot module mdot_tnt.rounding module

mdot_tnt.truncated_newton module module mdot_tnt.batched mdot_tnt.lowmem mdot_tnt.mdot mdot_tnt.rounding mdot_tnt.truncated_newton

N

newton_solve() (mdot_tnt.lowmem.LowMemoryTruncatedNewtonProjector method) (mdot_tnt.truncated_newton.TruncatedNewtonProjector method)

P

preprocess_marginals() (in module mdot_tnt.mdot) project() (mdot_tnt.batched.BatchedTruncatedNewtonProjector method) (mdot_tnt.lowmem.LowMemoryTruncatedNewtonProjector method) (mdot_tnt.truncated_newton.TruncatedNewtonProjector method)

R

round_altschuler() (in module mdot_tnt.rounding)

rounded_cost_altschuler() (in module mdot_tnt.rounding)

S

smooth_marginals() (in module mdot_tnt.mdot) solve_OT() (in module mdot_tnt) solve_OT_batched() (in module mdot_tnt) (in module mdot_tnt.batched)

solve_OT_lowmem() (in module mdot_tnt.lowmem), [1] squared_euclidean() (in module mdot_tnt.lowmem)

T

TruncatedNewtonProjector (class in mdot_tnt.truncated_newton)