jwtkeeble issues

Repositories
Issues
Comments

Results 2 issues of


                                            jwtkeeble

Is there a current work around for multiple devices?

Hi all, Firstly, I just want to say that this package is great! I was just wondering if there exists a current work around when using `@scan_tqdm` within `jax.experimental.shard_map` I...

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

Hi All, Is there a preferred way to visualize neural networks written within the `equinox` package? I understand that there's a pretty print for the PyTree representation. But is there...

question