Stars
A playbook for systematically maximizing the performance of deep learning models.
A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
[ICLR'25] Artificial Kuramoto Oscillatory Neurons
Rescure useful code in TensorNetwork by google for further personal usage.
Python optical flow visualization following Baker et al. (ICCV 2007) as used by the MPI-Sintel challenge
Unified representation for (differentiable) correspondences between objects in numpy and torch
nuvolos-cloud / PyMesh
Forked from PyMesh/PyMeshGeometry Processing Library for Python
A fast and robust total curvature estimation method, for both triangle meshes and point clouds.
Implementation of Complex Functional Maps
Implementing the Denoising Diffusion Probabilistic Model in Flax
The repository containing tools and information about the WoodScape dataset.
Auto-detect the connected display hardware and load the appropriate X11 setup using xrandr
Gradient, divergence, Laplacian on point clouds. Useful for setting up and solving PDEs on point clouds and learning anisotropic features in deep learning.
A lightweight 3D Morphable Face Model library in modern C++
Skin3D: Detection and Longitudinal Tracking of Pigmented Skin Lesions in 3D Total-Body Textured Meshes
KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Extended Convolution Histogram of Orientations
Steklov Spectral Geometry for Extrinsic Shape Analysis