ezjsiwu

Results 2 comments of ezjsiwu

hi there! im running into the same error with python 3.11 environment.. Anyone has figured out which version of jax is appropriate for this env?

hey @wd60622 ! thanks for the response.. I was able to create the plots just by extracting the get_channel_contributions_forward_pass_grid() function.. How can i submit a PR on this?