jax

Blog

Jax Image

What is JAX? Training Deep Learning models with JAX and Python

31st January 2024

Optimize your ML Kernels and speed up your AI training with JAX

jax Deep Learning Optimisation