JAX

High-performance automatic differentiation and compilation for ML research.

About JAX

Automatic differentiation and XLA compilation brought together for high-performance machine learning research.

Loading reviews...

omi

AI wearables for seamless voice transcription and communication.

deepwiki-open

Open Source AI Wiki Generator for Code Repositories

kubectl-ai

AI-powered Kubernetes assistant for streamlined cluster management.

AiToEarn

Use AI to boost your earnings effortlessly.

Get notified when JAX's pricing changes.