Home

Welcome to the hasegawa_wakatani wiki!

This repository contains scripts for simulating the Hasegawa-Wakatani turbulence model. Few Direct Numerical Simulation (DNS) algorithms are implemented for solving in 2D space or using a single poloidal mode.

The main dependencies are:

• Jax : accelerated array library
• Diffrax : numerical differential equation solvers
• Xarray : for saving the data simulation

Known issues

• neumann boundary make the findiff_2d simulation freeze
• when using GPU, you might face a out of memory error during plotting. Solution: plot with the CPU with python -m hasegawa_wakatani path_to_zarr.zarr --cpu
• if the simulation crashes with a out of memory error, you can try to force splitting the simulation with the command line: python -m hasegawa_wakatani path_to_yaml.yaml then call python -m hasegawa_wakatani path_to_zarr.zarr --tf TF with a increasing TF

References

• Ozgur Gurcan's repo
• Pierre Morel
• Other awesome projects: hw2d, PhiFlow