nixpkgs/pkgs/development/python-modules/flaxlib
..
default.nix