JAX Refs Enable Owned Memory And Mutation With XLA Optimizations · Digg