Tim Bechtel

icon location Working on Graph Neural Networks for Materials Science at Humboldt Universität zu Berlin

Results 2 comments of Tim Bechtel

VAE example outdated

As far as I can see `jax.random.PRNGKey` is still referenced in [Pseudo Random Numbers in JAX](https://jax.readthedocs.io/en/latest/jax-101/05-random-numbers.html). I also updated my JAX version to 0.4.23, and this had no problem with...

pad_with_graphs written with numpy

I was also wondering this in the past. Another piece where numpy is used/can be used is `batch`/`batch_np`. IIRC the numpy version was much faster in some situations, hinting that...