Stream

Quiz · 2 questions

🌳 Pytrees — JAX's Universal Data Structure

JAX's universal nested-structure abstraction

Level 0Curious

0 XP0/73 lessons0/17 achievements

0/100 XP to next level100 XP to go0% complete

Quiz

01What is a pytree?

02How do you make your custom dataclass usable as a pytree?

← PreviousPytrees in Practice: Neural Networks as Pytrees 🧠 Building Neural Networks — Flax, Equinox, and Friends →Why JAX Doesn't Include a Neural Network Library

Spotted a bug or have feedback on this page?Report an Issue

Comments 0

🔔 Reply notifications (sign in)

Sign in — Please sign in to comment.

No comments yet — be the first.