3 Commits

Author	SHA1	Message	Date
Pablo Montalvo	4afd3f4820	Model debugger upgrades (#37391) ... * debugging improvements * add debugging details * add more debugging details * debug more * clean up layers + output * add summary json file * cleanup * copies 👀 * remove hooks + add documentation * draft a small test, why not * respect the format (respect it) * fixup imports * nit * add tests and configurable pruning of layers	2025-04-18 16:45:54 +02:00
Joao Gante	949cca4061	[CI] doc builder without custom image (#36862) ... * no image * test * revert jax version updates * make fixup * update autodoc path for model_addition_debugger * shieldgemma2 * add missing pages to toctree	2025-03-21 09:10:27 +00:00
Pablo Montalvo	1d3f35f30a	Add model visual debugger (#36798) ... * draft of model tracer visualiser * add context manager in addition to decorator * add debug utils to init * move model debugging utils to dedicated file * add documentation * protect some imports * format * move and protect imports * format * doc: improve errors in case of broken dummy imports. * format * use automatic torch backend * update doc * fix backend * (TEMP) move to dummies while backend wait * update documentation * doc	2025-03-20 17:37:29 +01:00