Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W _ _args (nnsight.contexts.Tracer.Tracer attribute) (nnsight.models.NNsightModel.NNsight attribute) _call_iter (nnsight.envoy.Envoy attribute) _condition (nnsight.contexts.Conditional.Conditional attribute) _conditional_dict (nnsight.contexts.Conditional.ConditionalManager attribute) _conditional_stack (nnsight.contexts.Conditional.ConditionalManager attribute) _conditioned_nodes_dict (nnsight.contexts.Conditional.ConditionalManager attribute) _custom_model (nnsight.models.NNsightModel.NNsight attribute) _dispatched (nnsight.models.NNsightModel.NNsight attribute) _envoy (nnsight.models.NNsightModel.NNsight attribute) _fake_inputs (nnsight.envoy.Envoy attribute) _fake_outputs (nnsight.envoy.Envoy attribute) _graph (nnsight.contexts.Conditional.Conditional attribute) (nnsight.contexts.Tracer.Tracer attribute) _invoker (nnsight.contexts.Tracer.Tracer attribute) _invoker_inputs (nnsight.contexts.Tracer.Tracer attribute) _kwargs (nnsight.contexts.Tracer.Tracer attribute) (nnsight.models.NNsightModel.NNsight attribute) _model (nnsight.contexts.Tracer.Tracer attribute) (nnsight.models.NNsightModel.NNsight attribute) _model_key (nnsight.models.NNsightModel.NNsight attribute) _session (nnsight.models.NNsightModel.NNsight attribute) _tracer (nnsight.envoy.Envoy attribute) A add() (nnsight.intervention.InterventionProtocol class method) (nnsight.patching.Patcher method) (nnsight.tracing.Bridge.Bridge method) (nnsight.tracing.Graph.Graph method) (nnsight.tracing.protocols.ApplyModuleProtocol class method) (nnsight.tracing.protocols.BridgeProtocol class method) (nnsight.tracing.protocols.ConditionalProtocol class method) (nnsight.tracing.protocols.EarlyStopProtocol class method) (nnsight.tracing.protocols.GradProtocol class method) (nnsight.tracing.protocols.LocalBackendExecuteProtocol class method) (nnsight.tracing.protocols.LockProtocol class method) (nnsight.tracing.protocols.Protocol class method) (nnsight.tracing.protocols.SwapProtocol class method) (nnsight.tracing.protocols.UpdateProtocol class method) (nnsight.tracing.protocols.ValueProtocol class method) add_bridge_proxy() (nnsight.tracing.Bridge.Bridge method) add_conditioned_node() (nnsight.contexts.Conditional.ConditionalManager method) (nnsight.tracing.protocols.ConditionalProtocol class method) alive (nnsight.tracing.Graph.Graph attribute) ApiConfigModel (class in nnsight.schema.Config) AppConfigModel (class in nnsight.schema.Config) apply() (in module nnsight.util) (nnsight.contexts.GraphBasedContext.GraphBasedContext method) ApplyModuleProtocol (class in nnsight.tracing.protocols) arg_dependencies (nnsight.tracing.Node.Node attribute) args (nnsight.tracing.Node.Node attribute) attached() (nnsight.tracing.Node.Node method) attachments (nnsight.tracing.Graph.Graph attribute) automodel (nnsight.models.LanguageModel.LanguageModel attribute) B backend (nnsight.contexts.session.Session.Session attribute) BaseNNsightModel (class in nnsight.schema.format.types) bool() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) Bridge (class in nnsight.tracing.Bridge) bridge (nnsight.contexts.session.Session.Session attribute) bridge_backend_handle() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) bridged_nodes (nnsight.tracing.Bridge.Bridge attribute) BridgeProtocol (class in nnsight.tracing.protocols) BridgeProtocol.BridgeException bytearray() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) bytes() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) C check_for_dependencies() (in module nnsight.contexts) clean() (nnsight.tracing.Node.Node method) clear_edits() (nnsight.models.NNsightModel.NNsight method) complex() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) cond() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) cond_dependency (nnsight.tracing.Node.Node attribute) Conditional (class in nnsight.contexts.Conditional) ConditionalManager (class in nnsight.contexts.Conditional) ConditionalProtocol (class in nnsight.tracing.protocols) config (nnsight.models.LanguageModel.LanguageModel attribute) ConfigModel (class in nnsight.schema.Config) copy() (nnsight.tracing.Graph.Graph method) count() (nnsight.intervention.InterventionHandler method) create() (nnsight.tracing.Graph.Graph method) (nnsight.tracing.Node.Node method) D deregister() (nnsight.contexts.GraphBasedContext.GlobalTracingContext static method) destroy() (nnsight.tracing.Node.Node method) device (nnsight.intervention.InterventionProxy property) dict() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) DictModel (class in nnsight.schema.format.types) dispatch_model() (nnsight.models.NNsightModel.NNsight method) done() (nnsight.tracing.Node.Node method) dtype (nnsight.intervention.InterventionProxy property) E EarlyStopProtocol (class in nnsight.tracing.protocols) EarlyStopProtocol.EarlyStopException edit() (nnsight.models.NNsightModel.NNsight method) edit_backend_execute() (nnsight.contexts.Tracer.Tracer method) EllipsisModel (class in nnsight.schema.format.types) Envoy (class in nnsight.envoy) execute() (nnsight.tracing.Graph.Graph method) (nnsight.tracing.Node.Node method) (nnsight.tracing.protocols.ApplyModuleProtocol class method) (nnsight.tracing.protocols.BridgeProtocol class method) (nnsight.tracing.protocols.ConditionalProtocol class method) (nnsight.tracing.protocols.EarlyStopProtocol class method) (nnsight.tracing.protocols.GradProtocol class method) (nnsight.tracing.protocols.LocalBackendExecuteProtocol class method) (nnsight.tracing.protocols.Protocol class method) (nnsight.tracing.protocols.SwapProtocol class method) (nnsight.tracing.protocols.UpdateProtocol class method) (nnsight.tracing.protocols.ValueProtocol class method) executed() (nnsight.tracing.Node.Node method) exit() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) F fetch_attr() (in module nnsight.util) float() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) forward() (nnsight.util.WrapperModule method) fulfilled() (nnsight.tracing.Node.Node method) FunctionModel (class in nnsight.schema.format.types) FunctionWhitelistError G get() (nnsight.contexts.Conditional.ConditionalManager method) get_bridge() (nnsight.tracing.protocols.BridgeProtocol class method) get_bridge_proxy() (nnsight.tracing.Bridge.Bridge method) get_conditional() (nnsight.tracing.protocols.ConditionalProtocol class method) get_graph() (nnsight.tracing.Bridge.Bridge method) get_interventions() (nnsight.intervention.InterventionProtocol class method) get_module() (nnsight.tracing.protocols.ApplyModuleProtocol class method) get_swap() (nnsight.tracing.protocols.SwapProtocol class method) GlobalTracingContext (class in nnsight.contexts.GraphBasedContext) GlobalTracingContext.GlobalTracingExit (class in nnsight.contexts.GraphBasedContext) GlobalTracingContext.GlobalTracingTorchHandler (class in nnsight.contexts.GraphBasedContext) grad (nnsight.intervention.InterventionProxy property) GradProtocol (class in nnsight.tracing.protocols) Graph (class in nnsight.tracing.Graph) graph (nnsight.contexts.session.Session.Session attribute) (nnsight.tracing.Node.Node attribute) graph_stack (nnsight.tracing.Bridge.Bridge attribute) GraphBasedContext (class in nnsight.contexts.GraphBasedContext) GraphModel (class in nnsight.schema.format.types) H handles (nnsight.intervention.HookHandler attribute) has_bridge() (nnsight.tracing.protocols.BridgeProtocol class method) has_conditional() (nnsight.tracing.protocols.ConditionalProtocol class method) HookHandler (class in nnsight.intervention) I id_to_graph (nnsight.tracing.Bridge.Bridge attribute) increment() (nnsight.tracing.protocols.GradProtocol class method) input (nnsight.envoy.Envoy attribute) (nnsight.envoy.Envoy property) input_hook (nnsight.intervention.HookHandler attribute) inputs (nnsight.contexts.Invoker.Invoker attribute) (nnsight.envoy.Envoy property) int() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) interleave() (nnsight.models.NNsightModel.NNsight method) intervene() (nnsight.intervention.InterventionProtocol class method) InterventionHandler (class in nnsight.intervention) InterventionProtocol (class in nnsight.intervention) InterventionProxy (class in nnsight.intervention) invoke() (nnsight.contexts.Tracer.Tracer method) Invoker (class in nnsight.contexts.Invoker) is_node_conditioned() (nnsight.contexts.Conditional.ConditionalManager method) (nnsight.tracing.protocols.ConditionalProtocol class method) iter() (nnsight.contexts.session.Session.Session method) Iterator (class in nnsight.contexts.session.Iterator) IteratorModel (class in nnsight.schema.format.types) K kwargs (nnsight.contexts.Invoker.Invoker attribute) (nnsight.tracing.Node.Node attribute) L LanguageModel (class in nnsight.models.LanguageModel) LanguageModelProxy (class in nnsight.models.LanguageModel) list() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) listeners (nnsight.tracing.Node.Node attribute) ListModel (class in nnsight.schema.format.types) local_backend_execute() (nnsight.contexts.session.Session.Session method) (nnsight.contexts.Tracer.Tracer method) LocalBackendExecuteProtocol (class in nnsight.tracing.protocols) LockProtocol (class in nnsight.tracing.protocols) locks (nnsight.tracing.Bridge.Bridge attribute) log() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) M MetaDispatcher (class in nnsight.models.NNsightModel) model (nnsight.contexts.session.Session.Session attribute) (nnsight.intervention.HookHandler attribute) (nnsight.models.LanguageModel.LanguageModel attribute) model_computed_fields (nnsight.schema.Config.ApiConfigModel attribute) (nnsight.schema.Config.AppConfigModel attribute) (nnsight.schema.Config.ConfigModel attribute) (nnsight.schema.format.types.BaseNNsightModel attribute) (nnsight.schema.format.types.DictModel attribute) (nnsight.schema.format.types.EllipsisModel attribute) (nnsight.schema.format.types.FunctionModel attribute) (nnsight.schema.format.types.GraphModel attribute) (nnsight.schema.format.types.IteratorModel attribute) (nnsight.schema.format.types.ListModel attribute) (nnsight.schema.format.types.NodeModel attribute) (nnsight.schema.format.types.NodeModel.Reference attribute) (nnsight.schema.format.types.SessionModel attribute) (nnsight.schema.format.types.SliceModel attribute) (nnsight.schema.format.types.TensorModel attribute) (nnsight.schema.format.types.TracerModel attribute) (nnsight.schema.format.types.TupleModel attribute) (nnsight.schema.Request.RequestModel attribute) (nnsight.schema.Response.ResponseModel attribute) (nnsight.schema.Response.ResultModel attribute) model_config (nnsight.schema.Config.ApiConfigModel attribute) (nnsight.schema.Config.AppConfigModel attribute) (nnsight.schema.Config.ConfigModel attribute) (nnsight.schema.format.types.BaseNNsightModel attribute) (nnsight.schema.format.types.DictModel attribute) (nnsight.schema.format.types.EllipsisModel attribute) (nnsight.schema.format.types.FunctionModel attribute) (nnsight.schema.format.types.GraphModel attribute) (nnsight.schema.format.types.IteratorModel attribute) (nnsight.schema.format.types.ListModel attribute) (nnsight.schema.format.types.NodeModel attribute) (nnsight.schema.format.types.NodeModel.Reference attribute) (nnsight.schema.format.types.SessionModel attribute) (nnsight.schema.format.types.SliceModel attribute) (nnsight.schema.format.types.TensorModel attribute) (nnsight.schema.format.types.TracerModel attribute) (nnsight.schema.format.types.TupleModel attribute) (nnsight.schema.Request.RequestModel attribute) (nnsight.schema.Response.ResponseModel attribute) (nnsight.schema.Response.ResultModel attribute) model_fields (nnsight.schema.Config.ApiConfigModel attribute) (nnsight.schema.Config.AppConfigModel attribute) (nnsight.schema.Config.ConfigModel attribute) (nnsight.schema.format.types.BaseNNsightModel attribute) (nnsight.schema.format.types.DictModel attribute) (nnsight.schema.format.types.EllipsisModel attribute) (nnsight.schema.format.types.FunctionModel attribute) (nnsight.schema.format.types.GraphModel attribute) (nnsight.schema.format.types.IteratorModel attribute) (nnsight.schema.format.types.ListModel attribute) (nnsight.schema.format.types.NodeModel attribute) (nnsight.schema.format.types.NodeModel.Reference attribute) (nnsight.schema.format.types.SessionModel attribute) (nnsight.schema.format.types.SliceModel attribute) (nnsight.schema.format.types.TensorModel attribute) (nnsight.schema.format.types.TracerModel attribute) (nnsight.schema.format.types.TupleModel attribute) (nnsight.schema.Request.RequestModel attribute) (nnsight.schema.Response.ResponseModel attribute) (nnsight.schema.Response.ResultModel attribute) module nnsight.contexts nnsight.contexts.Conditional nnsight.contexts.GraphBasedContext nnsight.contexts.Invoker nnsight.contexts.session.Iterator nnsight.contexts.session.Session nnsight.contexts.Tracer nnsight.envoy nnsight.intervention nnsight.models nnsight.models.LanguageModel nnsight.models.NNsightModel nnsight.patching nnsight.schema nnsight.schema.Config nnsight.schema.format.functions nnsight.schema.format.types nnsight.schema.Request nnsight.schema.Response nnsight.tracing nnsight.tracing.Bridge nnsight.tracing.Graph nnsight.tracing.Node nnsight.tracing.protocols nnsight.tracing.Proxy nnsight.util modules (nnsight.intervention.HookHandler attribute) modules() (nnsight.envoy.Envoy method) N name (nnsight.tracing.Node.Node attribute) name_idx (nnsight.tracing.Graph.Graph attribute) named_modules() (nnsight.envoy.Envoy method) next() (nnsight.contexts.Tracer.Tracer method) NNsight (class in nnsight.models.NNsightModel) nnsight.contexts module nnsight.contexts.Conditional module nnsight.contexts.GraphBasedContext module nnsight.contexts.Invoker module nnsight.contexts.session.Iterator module nnsight.contexts.session.Session module nnsight.contexts.Tracer module nnsight.envoy module nnsight.intervention module nnsight.models module nnsight.models.LanguageModel module nnsight.models.NNsightModel module nnsight.patching module nnsight.schema module nnsight.schema.Config module nnsight.schema.format.functions module nnsight.schema.format.types module nnsight.schema.Request module nnsight.schema.Response module nnsight.tracing module nnsight.tracing.Bridge module nnsight.tracing.Graph module nnsight.tracing.Node module nnsight.tracing.protocols module nnsight.tracing.Proxy module nnsight.util module Node (class in nnsight.tracing.Node) node (nnsight.tracing.Proxy.Proxy attribute) NodeModel (class in nnsight.schema.format.types) NodeModel.Reference (class in nnsight.schema.format.types) nodes (nnsight.tracing.Graph.Graph attribute) O obj (nnsight.patching.Patch attribute) output (nnsight.envoy.Envoy attribute) (nnsight.envoy.Envoy property) output_hook (nnsight.intervention.HookHandler attribute) P parent (nnsight.patching.Patch attribute) Patch (class in nnsight.patching) patch() (nnsight.patching.Patch method) Patcher (class in nnsight.patching) patches (nnsight.patching.Patcher attribute) path (nnsight.envoy.Envoy attribute) peek() (nnsight.contexts.Conditional.ConditionalManager method) peek_conditional() (nnsight.tracing.protocols.ConditionalProtocol class method) peek_graph() (nnsight.tracing.Bridge.Bridge method) (nnsight.tracing.protocols.BridgeProtocol class method) pop() (nnsight.contexts.Conditional.ConditionalManager method) pop_conditional() (nnsight.tracing.protocols.ConditionalProtocol class method) pop_graph() (nnsight.tracing.Bridge.Bridge method) prepare_inputs() (nnsight.tracing.Node.Node class method) preprocess() (nnsight.tracing.Node.Node method) Protocol (class in nnsight.tracing.protocols) Proxy (class in nnsight.tracing.Proxy) proxy (nnsight.tracing.Node.Node attribute) proxy_class (nnsight.models.LanguageModel.LanguageModel attribute) (nnsight.models.NNsightModel.NNsight attribute) proxy_value (nnsight.tracing.Node.Node attribute) proxy_wrapper() (in module nnsight.tracing.Proxy) push() (nnsight.contexts.Conditional.ConditionalManager method) push_conditional() (nnsight.tracing.protocols.ConditionalProtocol class method) R redundant() (nnsight.tracing.Node.Node method) register() (nnsight.contexts.GraphBasedContext.GlobalTracingContext static method) remote_backend_get_model_key() (nnsight.contexts.session.Session.Session method) (nnsight.contexts.Tracer.Tracer method) remote_backend_handle_result_value() (nnsight.contexts.session.Session.Session method) (nnsight.contexts.Tracer.Tracer method) remote_backend_postprocess_result() (nnsight.contexts.session.Session.Session method) (nnsight.contexts.Tracer.Tracer method) replacement (nnsight.patching.Patch attribute) RequestModel (class in nnsight.schema.Request) reset() (nnsight.tracing.Graph.Graph method) (nnsight.tracing.Node.Node method) resolve_dependencies() (in module nnsight.contexts) ResponseModel (class in nnsight.schema.Response) ResponseModel.JobStatus (class in nnsight.schema.Response) restore() (nnsight.patching.Patch method) ResultModel (class in nnsight.schema.Response) S save() (nnsight.intervention.InterventionProxy method) scan (nnsight.contexts.Invoker.Invoker attribute) scan() (nnsight.models.NNsightModel.NNsight method) scanning (nnsight.contexts.Invoker.Invoker attribute) sequential (nnsight.tracing.Graph.Graph attribute) Session (class in nnsight.contexts.session.Session) session() (nnsight.models.NNsightModel.NNsight method) SessionModel (class in nnsight.schema.format.types) set() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) set_bridge() (nnsight.tracing.protocols.BridgeProtocol class method) set_module() (nnsight.tracing.protocols.ApplyModuleProtocol class method) set_value() (nnsight.tracing.Node.Node method) shape (nnsight.intervention.InterventionProxy property) SliceModel (class in nnsight.schema.format.types) stop() (nnsight.intervention.InterventionProxy method) str() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) style() (nnsight.intervention.InterventionProtocol class method) (nnsight.tracing.protocols.ApplyModuleProtocol class method) (nnsight.tracing.protocols.BridgeProtocol class method) (nnsight.tracing.protocols.ConditionalProtocol class method) (nnsight.tracing.protocols.EarlyStopProtocol class method) (nnsight.tracing.protocols.GradProtocol class method) (nnsight.tracing.protocols.LocalBackendExecuteProtocol class method) (nnsight.tracing.protocols.LockProtocol class method) (nnsight.tracing.protocols.Protocol class method) (nnsight.tracing.protocols.SwapProtocol class method) (nnsight.tracing.protocols.UpdateProtocol class method) (nnsight.tracing.protocols.ValueProtocol class method) SwapProtocol (class in nnsight.tracing.protocols) T t (nnsight.models.LanguageModel.LanguageModelProxy property) target (nnsight.tracing.Node.Node attribute) TensorModel (class in nnsight.schema.format.types) to() (nnsight.envoy.Envoy method) (nnsight.models.NNsightModel.NNsight method) token (nnsight.models.LanguageModel.LanguageModelProxy property) TokenIndexer (class in nnsight.models.LanguageModel) tokenizer (nnsight.models.LanguageModel.LanguageModel attribute) trace() (nnsight.models.NNsightModel.NNsight method) Tracer (class in nnsight.contexts.Tracer) tracer (nnsight.contexts.Invoker.Invoker attribute) TracerModel (class in nnsight.schema.format.types) try_deregister() (nnsight.contexts.GraphBasedContext.GlobalTracingContext static method) try_register() (nnsight.contexts.GraphBasedContext.GlobalTracingContext static method) tuple() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) TupleModel (class in nnsight.schema.format.types) U update() (nnsight.intervention.InterventionProxy method) UpdateProtocol (class in nnsight.tracing.protocols) V validate (nnsight.tracing.Graph.Graph attribute) value (nnsight.tracing.Node.Node attribute) (nnsight.tracing.Node.Node property) (nnsight.tracing.Proxy.Proxy property) ValueProtocol (class in nnsight.tracing.protocols) vis() (nnsight.contexts.GraphBasedContext.GraphBasedContext method) (nnsight.tracing.Graph.Graph method) visualize() (nnsight.tracing.Node.Node method) W wrap() (in module nnsight.util) WrapperModule (class in nnsight.util)
