ML researcher, JAX co-author
Pinned Loading
-
jax-ml/jax
jax-ml/jax PublicComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
rust-lang/rust
rust-lang/rust PublicEmpowering everyone to build reliable and efficient software.
-
google/jaxopt
google/jaxopt PublicHardware accelerated, batchable and differentiable optimizers in JAX.
-
openxla/xla
openxla/xla PublicA machine learning compiler for GPUs, CPUs, and ML accelerators
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.