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 shell
Note
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=false
to run the code because of a bug in JAX, see this discussion.