High Level Neural Network Modeling with Flax
Marshud
Command Palette
Search for a command to run...
Archive
8 posts published so far
Functional State Management and Low-Level Control Flow in JAX
Marshud
Managing PRNG Keys in JAX
Marshud
Program Transformations and PyTrees in JAX
Marshud
JAX Core Fundamentals: Pure Functional Arrays
Marshud
Why the JAX AI Stack is built for the Future of Foundation Models
Marshud
Deploying on Fridays?
Marshud
Welcome. Let’s all settle in, shall we?
Marshud