grow the AABB to include a point

Interpolate the attributes using the rasterized map Args: attr (Tensor, num_vertices 3, float): Attributes of the mesh

Calculate the area of the AABB

Utility function to clamp a value between a minimum and a maximum

Initialize AABB to an invalid state

Check if two AABBs overlap

Rasterize the UV coordinates to a barycentric coordinates & Triangle idxs texture map Args: uv (Tensor, num_vert

Create a triangle centroid

Align the mesh with the main axis Args: vertex_positions (Float[Tensor, "Nv 3"]): Vertex positions vertex_no

Assigns the face UV to the atlas index Args: vertex_positions (Float[Tensor, "Nv 3"]): Vertex positions tria

Assigns each vertex to a cube face based on the face normal Args: vertex_positions (Tensor, Nv 3, float): Vertex positio

Calculate the tangents for each triangle Args: vertex_positions (Float[Tensor, "Nv 3"]): Vertex positions ve

Distribute the individual UVs in the atlas Args: face_uv (Float[Tensor, "Nf 3 2"]): Face UV coordinates assi

Find the slice offset and scale Args: index (Integer[Tensor, "Nf"]): Atlas index Returns: Float[Ten

Get the unique face UV Args: uv (Float[Tensor, "Nf 3 2"]): UV coordinates Returns: Float[Tensor, "U

Handle the remaining UVs (The ones that are not slices) Args: uv (Float[Tensor, "Nf 3 2"]): UV coordinates i

Handle the slice UVs Args: uv (Float[Tensor, "Nf 3 2"]): UV coordinates index (Integer[Tensor, "Nf"]): Atlas

Rotate the UV slices so they are in a consistent space Args: vertex_positions (Float[Tensor, "Nv 3"]): Vertex positions

Function to check if a point (xy) is inside a triangle defined by vertices v1, v2, v3

Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). Comment by Ross Wightman: This is the same as the D