XLA

JAXgoogle/jax

A Python library for composable transformations of numerical programs: automatic differentiation, vectorization, and JIT compilation to GPU/TPU.

JAX GitHub Discussionsgoogle/jax

A Python library for composable transformations of numerical programs: automatic differentiation, vectorization, and JIT compilation to GPU/TPU.

jaxjax-ml/jax

A Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and machine learning.

Swift for Tensorflowtensorflow/swift

An archived experiment integrating TensorFlow's machine learning capabilities directly into the Swift programming language with first-class differentiable programming.

Nxelixir-nx/nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir, enabling machine learning and scientific computing.

Neural Tangentsgoogle/neural-tangents

A high-level neural network API for specifying and analyzing infinite-width neural networks as Gaussian Processes in Python.