This repository contains the code of the notebooks presented on my website https://juanitorduz.github.io.
-
Feel free to share and use the code.
-
If you find a mistake or typo please let me know (ideally report as an issue or open a pull request).
-
Comments and suggestions are always welcome.
We use pixi, see It's time to try out pixi! to get started.
pixi shellNote
The nutpie package requires cargo, a rust package manager. For installation instructions, see here.
Important
We need to set the environment variable
XLA_FLAGS=--xla_cpu_use_thunk_runtime=falseto run the code because of a bug in JAX, see this discussion.