I maintain jax4j, an Apache-2.0 tensor library for Java. This notebook lets you edit Java code and compare CPU and browser WebGPU results without installing a JDK.

Source: https://github.com/nablatensor-dev/jax4j

I’d welcome feedback on the API and reports of browser/GPU compatibility.