Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as higher level abstraction for domain experts

Improvements to KANPRO.

Computations and statistics on manifolds with geometric structures.

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeX…

PyTorch implementations of Learning Mesh-based Simulation With Graph Networks

Deep and online learning with spiking neural networks in Python

Deep Learning library for Lava

A Software Framework for Neuromorphic Computing

Monotone Parameterization Toolkit (MParT): A core library for constructing and using transport maps.

V2E: From video frames to DVS events

A deep learning library for spiking neural networks which is based on PyTorch, focuses on fast training and supports inference on neuromorphic hardware.

ICNS Event Based Camera Simulator

Newton and Quasi-Newton optimization with PyTorch

Rust bindings for the C++ api of PyTorch.

A playbook for systematically maximizing the performance of deep learning models.

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

PyDEC: A Python Library for Discretizations of Exterior Calculus for simplicial complexes of any dimension embedded or not and for cubical complexes of any dimension. Implements discrete exterior d…

python version of the No-U-Turn Sampler (NUTS) from Hoffman & Gelman, 2011

Learning the optimal transport map via input convex neural neworks

3D Roto-translation equivariant convolutions