Cheatsheet

JAX Tips and Tricks

This is an unordered list of useful things I found while using JAX, that don’t seem to be documented well somewhere else. Partially this is because these features are experimental, so don’t depend on them working the exact same way in future versions of JAX. Nevertheless, they might be useful during development and testing and have been tested using JAX 0.5.3 as of writing.

VSC/Slurm/Spack Tips and Tricks

This is not official documentation for the Vienna Scientific Cluster. For this check the VSC Docs. Instead, this is my personal cheat sheet of things that are not well documented elsewhere. Also while the content is focused on the VSC, most of the things mentioned here also apply to similar setups that use Slurm at other universities.