-
Peter Hawkins authored
JAX will soon import scipy lazily, so code that relies on JAX having already imported submodules of scipy will fail. Change code to import what it uses. PiperOrigin-RevId: 480841853 Change-Id: I18a744a561d73da3ccd43c26fcd99d10070c058e
a31061e8
