Closed olupton closed 1 week ago
Maybe mention in the documentation that now there is a section about compiletime and one about heap memory usage?
This actually landed in https://github.com/NVIDIA/JAX-Toolbox/pull/926, I'll push a doc update separately soon.
Maybe mention in the documentation that now there is a section about compiletime and one about heap memory usage?