Skip to content

Framework

JAX

JAX is Google's Python library for high-performance numerical computing and machine learning, using composable function transforms and XLA compilation.

Known aliases

  • JAX
  • jax[cuda13]

Current stories