Jax
-
-
Unleashing the Gradient: How JAX Makes Automatic Differentiation Feel Like Magic
12 min read -
Accelerating AI/ML Model Training with Custom Operators – Part 3.A
13 min read -
Part 1: Build 124M GPT2 with JAX. Part 2: Optimize the training speed in Single…
8 min read -
Everything you need to assemble the DQN Megazord in JAX.
20 min read -
All hail the return of Keras
17 min read -
Hit the road to super-fast AI/ML development
12 min read -
All the tricks and details you wish you knew about PPO
9 min read -
Solving the CartPole environment with DQN in under a second
11 min read