maxxxzdn

UPT

Code for the paper Universal Physics Transformers

EagleMeshTransformer

Code for Mesh Transformer describes in the EAGLE dataset

minified-generative-models

Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc

swirl-fem

Stable-Neural-SDEs

perceiver-pytorch

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

pytorch_memlab

Profiling and inspecting memory in pytorch

AIGLETools

Multi-dimensional (generalized) Langevin equation

ckconv

Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/2102.02611

conditional-flow-matching

TorchCFM: a Conditional Flow Matching library

picograd

Rust Implementation of micrograd

system-of-N-coupled-oscillators

University laboratory work: The dynamics of of N-coupled oscillators

Brownian-dynamics-in-a-time-varying-force-field

A python code to calculate the Brownian motion of colloidal particles in a time varying force field.

pybbfmm

Black-box fast multipole method package for Python

fmm

Fast Multipole Method in Python

pycharge

Electrodynamics simulator for calculating the fields and potentials generated by moving point charges and simulating oscillating dipoles.

pytreegrav

Fast N-body gravitational force and potential in Python

eqnn-jax

Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.

Triton-Puzzles

Puzzles for learning Triton

JaxLightning

Running Jax in PyTorch Lightning

LovelyPlots

Matplotlib style sheets to nicely format figures for scientific papers, thesis and presentations while keeping them fully editable in Adobe Illustrator.

escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

pinn-torch

PINN (Physics-Informed Neural Networks) on Navier-Stokes Equations

fast-simplification

Fast Quadratic Mesh Simplification

LPSDA

Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"

Steerable-E3-GNN

E(3) Steerable Graph Neural Network

ai-deadlines

:alarm_clock: AI conference deadline countdowns

exmol

Explainer for black box models that predict molecule properties

dmol-book

Deep learning for molecules and materials book

variational-zoo

Variational inference and disentangled representations through unsupervised learning