flax
ca72d0a9 - Do not call `jnp` on shapes in AutoEncoder example

Commit

4 years ago

Do not call `jnp` on shapes in AutoEncoder example Otherwise can't jit/abstact-eval `model.apply`

References

Author

romanngg

Parents