jax-js is a JavaScript library that brings JAX-style machine learning capabilities to the browser. It allows users to perform high-performance numerical computations using familiar NumPy-like syntax and runs entirely client-side. The framework supports GPU acceleration through WebGPU and offers features like automatic differentiation and JIT compilation.

A particle life simulation is developed in a browser using the WebGPU API, showcasing life-like behaviors through an unconventional model where asymmetric forces between particles mimic predator-prey interactions. The article details the physics behind the simulation, the implementation using WebGPU, and the advantages of this modern API over traditional graphics programming methods.