jaxnet icon indicating copy to clipboard operation
jaxnet copied to clipboard

Published 20 hours ago • verified publisher icon juliuskunze
→

Metadata

Concise deep learning for JAX

Results 2 jaxnet issues
Sort by recently updated
recently updated
newest added

PRNG handling akin to parameters

2 comment

Handling parameters in JAX can get annoying, but what really concerns me even more is handling PRNG keys. JAX has a done a lot of great work to build a...

samuela avatar samuela

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

See `TODO` in `test_parametrized_jit`.

juliuskunze avatar juliuskunze

← Metadata

183
Stars
14
Forks
Watchers

Owner

verified publisher icon juliuskunze

Metadata

Concise deep learning for JAX

Back