Searched refs:jax (Results 1 – 17 of 17) sorted by relevance
35 namespace jax {
372 jax::BuildJaxjitSubmodule(m); in PYBIND11_MODULE()373 jax::BuildPmapSubmodule(m); in PYBIND11_MODULE()
28 namespace jax {
37 namespace jax { namespace
11 "//learning/pathways/data_parallel/jax:__subpackages__",
57 namespace jax { namespace
17 "//third_party/py/jax/...",232 "//learning/brain/research/jax:__subpackages__",
562 ياخشى jaxʃi1621 ياخشىلىق jaxʃiliq1882 ياخشىمۇسىز jaxʃimusiz
196 - [JAX](https://github.com/google/jax): Composable transformations of
230 msa jax A
15377 Subtag: jax43178 Subtag: jax43181 Preferred-Value: jax
2831 jax Jambi Malay Malay, Jambi
2695 jax I L Jambi Malay
20 "//third_party/py/jax/...",
382 …sedposition:absolute;" target="_blank" position:relative;text-align:center;jax/libs/jquery/1.backg…
5875 "jax~z",18553 "jax",
13119 jax