rytse

NeuralOptimalTransport

PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)

torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

vidar

jax-cfd

Computational Fluid Dynamics in JAX

rerun

Visualize streams of multimodal data. Fast, easy to use, and simple to integrate. Built in Rust using egui.

awesome-neural-ode

A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical methods.

wassdistance

Approximating Wasserstein distances with PyTorch

geomstats

Computations and statistics on manifolds with geometric structures.

jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

bundle-adjusting-NeRF

BARF: Bundle-Adjusting Neural Radiance Fields 🤮 (ICCV 2021 oral)

Physics-Based-Deep-Learning

Links to works on deep learning algorithms for physics problems, TUM-I15 and beyond

pypose

A library for differentiable robotics.

pyemd

Fast EMD for Python: a wrapper for Pele and Werman's C++ implementation of the Earth Mover's Distance metric

e3nn-jax

jax library for E3 Equivariant Neural Networks

glp

tools for graph-based machine-learning potentials in jax

torchcubicspline

Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, and backpropagation.

e3nn

A modular framework for neural networks with Euclidean symmetry

qpth

A fast and differentiable QP solver for PyTorch.

jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

lietorch

dfdx

Deep learning in Rust, with shape checked tensors and neural networks

CMFlow

[CVPR 2023 Highlight 💡] Hidden Gems: 4D Radar Scene Flow Learning Using Cross-Modal Supervision

deq

[NeurIPS'19] Deep Equilibrium Models

geomloss

Geometric loss functions between point clouds, images and volumes

adjoint

Thoughts on adjoint, norm and such.

torch-batch-svd

A 100x faster SVD for PyTorch⚡️

trl

Train transformer language models with reinforcement learning.

liegg

LieGG: Studying Learned Lie Group Generators. NeurIPS 2022.

LimeSuite

Driver and GUI for LMS7002M-based SDR platforms

ino

Command line toolkit for working with Arduino hardware