Paul Wiemann

my current understanding is that buffer donation works on TPUs and GPUs (since jax 0.1.73) but not on CPUs. are there plans to support CPUs as well?

i wonder if this a good time to create the contrib module and place distreg there

The name `Const` would suggest that the value is constant and immutable. From the two options, I like `Value` better.