https://github.com/mark1626/finitevolume-jax

Finite volume simulation with JAX
https://github.com/mark1626/finitevolume-jax

jax physics simulation

Last synced: 25 days ago
JSON representation

Finite volume simulation with JAX

• Host: GitHub
• URL: https://github.com/mark1626/finitevolume-jax
• Owner: Mark1626
• License: gpl-3.0
• Created: 2023-12-13T16:14:48.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-12-26T06:14:07.000Z (over 2 years ago)
• Last Synced: 2025-12-25T18:46:00.725Z (6 months ago)
• Topics: jax, physics, simulation
• Language: Jupyter Notebook
• Homepage:
• Size: 281 KB
• Stars: 3
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# JAX Finite Volume Simulation

The jupyter notebook contains a hardware accelerated finite volume simulation made using JAX.

You can read the write up about this in the article I wrote on [Medium](https://medium.com/@nimalan854/building-a-hardware-accelerated-simulation-using-jax-00437d562002)

The code is a port of the python tutorial made by Philip Mocz, https://github.com/pmocz/finitevolume-python/

![finite_volume_sim.gif](./finite_volume_sim.gif)

## Trying this out

Google Collab provides a GPU runtime, you can try this out using [this link](https://colab.research.google.com/github/Mark1626/finitevolume-jax/blob/main/Finite_Volume_Fluid_Simulation_in_JAX.ipynb)