zongyi-li

VAR

[GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-simple, user-friendly yet state-of-the-art* codebase for autoregressive image generation!

neuraloperator

Learning in infinite dimension with neural operators.

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

GaLore

GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection

LeanCopilot

LLMs as Copilots for Theorem Proving in Lean

modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

deep_complex_networks

Implementation related to the Deep Complex Networks

PointFlow

PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows

Neural-PDE-Solver

fft-conv-pytorch

Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.

modulus-makani

Massively parallel training of machine-learning based weather and climate models

CAMLab-DLSCTutorials

MFC

Exascale simulation of multiphase/physics fluid dynamics

CFDBench

A large-scale benchmark for machine learning methods in fluid dynamics

poseidon

Code for the paper "Poseidon: Efficient Foundation Models for PDEs"

TOTEM

The official code 👩‍💻 for - TOTEM: TOkenized Time Series EMbeddings for General Time Series Analysis

Equivariant-Net

DPOT

Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"

HierarchicalDesign

A computational building block approach towards multiscale architected materials analysis and design with application to hierarchical metal metamaterials

controlgym

Controlgym: Large-Scale Control Environments for Benchmarking Reinforcement Learning Algorithms

CST

Continuous Spatiotemporal Transformer

fourier-neural-mappings

An extension of Fourier Neural Operator to finite-dimensional input and/or output spaces.

cvit

SPFNO

UnifiedPDESolvers

torch-local-neural-operators

acm270_projects

Geo-FNO-catheter

FNO_Multi_Blobs

Fourier Neural Operator Modelling for Multi-Blob Diffusion within a Reduced MHD case built using JOREK

Propeller