Hackernews posts about Jax

Jax is an open-source Python library developed by Google that provides automatic differentiation and XLA (Accelerated Linear Algebra) for efficient execution of machine learning models on CPUs and GPUs.

Related: PyTorch   TensorFlow   Julia   Keras  
  1. Speeding up my Ray Tracer using JAX (kayleegeorge.github.io)
  2. How to Think About GPUs (jax-ml.github.io)
  3. Japan's Creepiest Station (www.tokyocowboy.co)