Cache miss breaks jit(parametrized(jit).apply) - Githubissues

juliuskunze / jaxnet

Concise deep learning for JAX

Apache License 2.0

184 stars 14 forks source link

Cache miss breaks jit(parametrized(jit).apply) #18

Open juliuskunze opened 4 years ago

juliuskunze commented 4 years ago

See TODO in test_parametrized_jit.