Andrea Schioppa

### Feature request Request for writable temporary directory handling within Python tests ### Motivation I'm currently using the safetensors library in a project that leverages the Bazel build system. Bazel's...

Is it possible to have early stopping in `lax.scan`?

7

I am considering a case in which there is a loop of operations, each one of them being expensive; one can bound the worst case number of operations with a...