Searched refs:jax_core (Results 1 – 1 of 1) sorted by relevance
1247 static const auto* jax_core = in BuildJaxjitSubmodule() local1250 new py::handle(jax_core->attr("ShapedArray")); in BuildJaxjitSubmodule()