Getting Started
Tutorials
Reference Documentation
Extending NetKet
API documentation
Developer Documentation
Behaves like jax.vmap but uses scan to chunk the computations in smaller chunks.
f (Callable) –
chunk_size (Optional[int]) –