Google Brain:
JAX (2020.9 - now), Tensorflow (2018.7-2020.8)
-
Google
- Mountain View, CA
- www.linkedin.com/in/qiaozhanglinkedin
- @zhangqiaorjc
Block or Report
Block or report zhangqiaorjc
Report abuse
Contact GitHub support about this user’s behavior. Learn more about reporting abuse.
Report abusePinned Loading
-
google/jax
google/jax PublicComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
-
google/paxml
google/paxml PublicPax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…
-
-
tensorflow/runtime
tensorflow/runtime PublicA performant and modular runtime for TensorFlow
-
spmd_pipeline_example.py
spmd_pipeline_example.py 1import itertools as it
23import jax
4import jax.numpy as jnp
5
Something went wrong, please refresh the page to try again.
If the problem persists, check the GitHub status page or contact support.
If the problem persists, check the GitHub status page or contact support.