tenhoeve13

Results 1 comments of tenhoeve13

jax version error

I'm still getting the error: AttributeError: module 'jax.interpreters.pxla' has no attribute 'ShardedDeviceArray' after trying the fix from @scploeger. Any help would be appreciated.