Hardware
The power of JAX
The power of JAX 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. Valid
The power of JAX 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.
Related
- Foundry: Template-Based CUDA Graph Context Materialization for Fast LLM Serving Cold Start
- Cut Checkpoint Costs with About 30 Lines of Python and NVIDIA nvCOMP
- New course: Efficient Inference with SGLang: Text and Image Generation, built in partnership with LMSys @lmsysorg and RadixArk @radixark, an…
- DynLP: Parallel Dynamic Batch Update for Label Propagation in Semi-Supervised Learning
Source: Francois Chollet (X) | 2026-04-10