C.W.K.
Stream
Quiz · 2 questions

Advanced Transformations and Patterns

custom_vjp, remat, scan, and meta-learning patterns

Level 0Curious
0 XP0/73 lessons0/17 achievements
0/100 XP to next level100 XP to go0% complete

Quiz

01When would you reach for jax.custom_vjp?
02What problem does jax.checkpoint (gradient checkpointing) solve?
Spotted a bug or have feedback on this page?Report an Issue

Comments 0

🔔 Reply notifications (sign in)
Sign inPlease sign in to comment.

No comments yet — be the first.