jax.live_arrays

Contents

jax.live_arrays#

jax.live_arrays(platform=None)[source]#

Return all live arrays in the backend for platform.

If platform is None, it is the default backend.