JAX iconJAX

oss Free Star35k

NumPy-like framework for composable function transformations with automatic differentiation, JIT compilation, and GPU/TPU acceleration

—
46.5+ Users
— Rating
— Founded

Overview

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more - jax-ml/jax

The Verdict

Who Should Use JAX?

Best For

  • [Add best use case 1]
  • [Add best use case 2]
  • [Add best use case 3]

Not Ideal For

  • [Add limitation 1]
  • [Add limitation 2]

What's Great

  • NotificationsYou must be signed in to change notification settings
  • Fork3.6k
  • Star35.7k
  • Transformations
  • Scaling
Official Site

Watch Out For

  • [Research G2/Capterra for cons]
  • [Add con 2]
G2 Reviews

Pricing

Open Source
$0
Apache License 2.0 - free and open source
View all features & details

Key Features

  • NotificationsYou must be signed in to change notification settings
  • Fork3.6k
  • Star35.7k
  • Transformations
  • Scaling
  • Current gotchas

Platforms

  • [Add supported platforms]

How It Compares

Feature JAX Competitor 1 Competitor 2
Key Feature — — —
Pricing — — —
Best For — — —

User Reviews

Loading reviews...