OctFormer: Octree-based Transformers for 3D Point Clouds

The Open Source AI-Powered Code Editor. A fork of VSCode and Continue and PearAI

LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite

Numpy implementation of Hilbert curves in arbitrary dimensions

Code for the paper Universal Physics Transformers

Code for Mesh Transformer describes in the EAGLE dataset

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

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

Profiling and inspecting memory in pytorch

Multi-dimensional (generalized) Langevin equation

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

TorchCFM: a Conditional Flow Matching library

Rust Implementation of micrograd

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

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

Black-box fast multipole method package for Python

Fast Multipole Method in Python

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

Fast N-body gravitational force and potential in Python

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

Puzzles for learning Triton

Running Jax in PyTorch Lightning

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

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

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

Fast Quadratic Mesh Simplification

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

E(3) Steerable Graph Neural Network