Reference implementation for the Hasegawa-Wakatani model of plasma turbulence inside nuclear fusion reactors in two dimensions

TORAX: Tokamak transport simulation in JAX

A Comprehensive Survey of Scientific Large Language Models and Their Applications in Scientific Discovery

Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"

Control of 2D Rayleigh Benard Convection using Deep Reinforcement Learning with Tensorforce and Shenfun.

The code used, and a docker image to run it, of the paper `Exploiting locality and physical invariants to design effective Deep Reinforcement Learning control of the unstable falling liquid film`

1D-1V Vlasov-Poisson(-Fokker-Planck), Plasma Physics PDE Simulation Tool in NumPy and experiment management in MLFlow

WarpX is an advanced electromagnetic & electrostatic Particle-In-Cell code.

Particle-in-cell code for plasma physics simulations

Experiment code for testing effect of various action space transformations in reinforcement learning

PyTorch implementation of Soft Actor-Critic (SAC), Twin Delayed DDPG (TD3), Actor-Critic (AC/A2C), Proximal Policy Optimization (PPO), QT-Opt, PointNet..

High-order Lagrangian Hydrodynamics Miniapp

The open-source tool for building high-quality datasets and computer vision models

Repository of Transformer based PyTorch Time Series Models

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008

The GitHub repository for the paper "Informer" accepted by AAAI 2021.

Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.

Lightweight, general, scalable C++ library for finite element methods

Deep Learning for Weather Forecasting, accepted applied data science of KDD 2019

[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis

The Electricity Transformer dataset is collected to support the further investigation on the long sequence forecasting problem.

🌊 Julia software for fast, friendly, flexible, ocean-flavored fluid dynamics on CPUs and GPUs

A Python library for amortized Bayesian workflows using generative neural networks.

Robust neural network surrogate for inertial confinement fusion

Python Package for Statistical Analysis of Turbulence Data

Statistics of Turbulence Python Package

PyTorch implementation of normalizing flow models