Go Back
Report Abuse

JAX

Description

A numerical computing library developed by Google, JAX enables high-performance machine learning research with automatic differentiation and GPU/TPU support.

Features

Automatic differentiation for complex functions.
Just-In-Time (JIT) compilation for optimized performance.
Support for parallelization and vectorization.