Quiz · 2 questions
⚡ jit — Just-In-Time Compilation with XLA
Just-In-Time compilation with XLA
Level 0Curious
0 XP0/73 lessons0/17 achievements
0/100 XP to next level100 XP to go0% complete
Quiz
01What does jax.jit actually do on the first call vs subsequent calls?
02You call a jit'd function with three different string values for an arg. JAX recompiles three times. What's the fix?
Comments 0
🔔 Reply notifications (sign in)Sign in — Please sign in to comment.
No comments yet — be the first.