nx
nx copied to clipboard
Default integers to 32-bit precision
This way tensors require less memory by default and people can opt-in if they work with really big numbers. We already default to 32-bit precision for floats. Sidenote: Jax does that too and even guards 64-bit precision behind a configuration flag.