{ pkgs, ... }:
let
  native_dependencies = with pkgs.python312Packages; [
    torch-bin
    torchvision-bin
    aggdraw # for visualtorch
    numpy
    scipy
    matplotlib
  ];
  tools = with pkgs; [
    ruff
  ];
in
{
  packages = native_dependencies ++ tools;
  languages.python = {
    enable = true;
    package = pkgs.python312;
    uv = {
      enable = true;
      sync.enable = true;
    };
    venv.enable = true;
  };

}