3 projects
neurarch-slurm
A SLURM cli_filter plugin that sizes a training job's model at submit, before it takes a reservation: refuses or warns when the requested GPU cannot hold the training peak, and warns when it asks for far more.
neurarch-skypilot
A SkyPilot admin policy that sizes a training job's model before it takes a GPU: refuses a request that cannot hold it, right-sizes one that asks for far more, and enforces a cost cap.
neurarch-trace
Run one forward pass over a PyTorch model and write a .neurarch.json graph with real per-layer shapes, for neurarch-mcp.