Any solutions for svi.init taking up so much gpu / cpu memory? Especially when i use more complex guides
You can use this tool to profile Profiling device memory — JAX documentation Please let us know if there are any places that we can improve in the svi.init method (based on the profiling result).