Viewing a single comment thread. View all comments

xl0 OP t1_ivj5slm wrote on November 8, 2022 at 10:32 AM

I started working on it. Will make sure repr works inside jit and parallel before moving to other things.

Please let me know if you have any thoughts, I'm very new to JAX.