AI Dynamics

Global AI News Aggregator

JAX Solver for Gyrokinetics Achieves 10x Speedup with CUDA

The power of JAX Eric Volkmann (@e_volkmann) Introducing gyaradax 🐉: A JAX solver for local flux-tube gyrokinetics with custom CUDA kernels for acceleration. This entire code was vibecoded by @ggalletti_ and me in a month. Validated against GKW (CPU-only Fortran code) with 10x speedups. Details and code in the replies. — https://nitter.net/e_volkmann/status/2041853935430881771#m

→ View original post on X — @fchollet, 2026-04-10 05:07 UTC

Commentaires

Leave a Reply

Your email address will not be published. Required fields are marked *