get out of dependency hell
Goal: reproduceable dev environments. Current state: python 3.10/3.11 reqd for some deps, installed on converging pc, but jaxlib w/ cuda only global on 3.8, no easy jaxlib w/ cuda via venv.
-
Check opt_tools can do jaxlibWithCuda, jax imports w GPU. Update lock -
Pull flake.nix from opt_tools up to encoder dynamics -
Test enc_dyn things on new install -
add docu / update readme enc_dyn
-
Check the parallel compilation / fwd compatibility issue or update nvidia driver
Edited by Kevin Haninger