import plotly.express as px def plot_curvature(curv): layers=list(curv.keys()) vals=list(curv.values()) fig=px.bar(x=layers,y=vals,title="Curvature per Layer") return fig def plot_precision_plan(plan): layers=list(plan.keys()) vals=list(plan.values()) fig=px.bar(x=layers,y=[1 if v=='fp16' else 0 for v in vals], title="Precision Plan (1=FP16, 0=FP32)") return fig