Getting Started
Tutorials
Reference Documentation
Extending NetKet
API documentation
Developer Documentation
Utility functions and classes.
HashableArray
This class wraps a numpy or jax array in order to make it hashable and equality comparable (which is necessary since a well-defined hashable object needs to satisfy obj1 == obj2 whenever hash(obj1) == hash(obj2).
obj1 == obj2
hash(obj1) == hash(obj2)
mpi.available
bool(x) -> bool
mpi.n_nodes
int([x]) -> integer int(x, base=10) -> integer
mpi.rank
mpi.MPI_jax_comm
mpi.MPI_py_comm