search

juliuskunze / jaxnet

Concise deep learning for JAX
Apache License 2.0
184 stars 14 forks source link

Allow lax.scan with unparametrized cell within parametrized function #15

Closed juliuskunze closed 4 years ago

juliuskunze commented 4 years ago

Fix test_scan_unparametrized_cell