jaxdem.utils.linalg# Utility functions to help with linear algebra. Functions unit(v) Normalize vectors along the last axis. jaxdem.utils.linalg.unit(v: Array) → Array[source][source]# Normalize vectors along the last axis. v: (…, D) returns: (…, D), unit vectors; zeros map to zeros.