Max Zhdanov's starred repositories
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
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
conditional-flow-matching
TorchCFM: a Conditional Flow Matching library
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.
pytreegrav
Fast N-body gravitational force and potential in Python
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.
pinn-torch
PINN (Physics-Informed Neural Networks) on Navier-Stokes Equations
fast-simplification
Fast Quadratic Mesh Simplification
Steerable-E3-GNN
E(3) Steerable Graph Neural Network
ai-deadlines
:alarm_clock: AI conference deadline countdowns
variational-zoo
Variational inference and disentangled representations through unsupervised learning