Realistic, physically accurate star cluster simulator with JAX and CUDA

This is a project I built years ago, when I was a student, now rebuilt with JAX and Python, using GPU acceleration. It runs a physically accurate simulation of a star cluster. All aspects follow physical law: position, mass and temperature of stars, relationships between temperature, radius, and luminosity, and even the color accurately reflects the temperature.

https://florin-andrei.medium.com/realistic-star-cluster-simulator-gpu-accelerated-using-jax-and-python-fadce3c71594

Code:

Demo on YouTube:

Using a regular gaming GPU you could do 10…20k stars, maybe more on a flagship model. But with the DGX Spark you can run the simulation with more than 100k stars. It’s very nice to have a lot of shared memory.

3 Likes