jwtkeeble comments

Repositories
Issues
Comments

Results 4 comments of


                                            jwtkeeble

Is there a current work around for multiple devices?

I can work on a simple code example to reproduce this error, and I'll share it down once it's written.

Is there a current work around for multiple devices?

So, I've managed to get a code snippet that works for a single device (but not for multiple when run over `jax.experimental.shard_map`), it doesn't give that exact same error as...

Is there a current work around for multiple devices?

Hi @zombie-einstein, no need to apologise! All's good. My current work around at the moment is to use the `logging` library and use an `io_callback` to print an estimate of...

[Question] Is there a preferred way to visualise a network written as an eqx.Module?

Thanks for the quick response, @patrick-kidger ! I'll have a look and see what I find and if I find anything useful, I'll attach it here.