JAX-CFD

Machine Learning-accelerated Computational Fluid Dynamics (CFD)

Datasets

The training data and evaluation data are in /global/cfs/cdirs/m3898/zhiqings/cfd/content

The trained models are in /global/cfs/cdirs/m3898/zhiqings/cfd/models

conda env create -f cfd.yml
conda activate cfd
git clone https://github.com/google/jax-cfd.git
cd jax-cfd
pip install -e ".[complete]"

All the scripts for reproduction are in scripts directory. For visualization, check plot_fig*.py files in the main directory.

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
data		data
figs		figs
logs		logs
models		models
new_figs		new_figs
samples		samples
scripts		scripts
.gitignore		.gitignore
README.md		README.md
cfd.yml		cfd.yml
plot_fig1.py		plot_fig1.py
plot_fig2.py		plot_fig2.py
plot_fig3.py		plot_fig3.py