Also, inspired by simplification you've done in SessionContext, maybe we should add companion object to FunctionFactory so it can register configuration as well. At the moment its a bit of work to register configuration:

https://github.com/milenkovicm/torchfusion/blob/634c2a4e39b81e1969671db8faea50b96bc43c7f/src/lib.rs#L101

let mut session_config = SessionConfig::new().with_information_schema(true);

session_config
      .options_mut()
      .extensions
      // register torch factory configuration
      .insert(TorchConfig::default());

it would be much better if config could be registered when ctx.with_function_factory(...) called ... but that could be a follow up PR

Originally posted by @milenkovicm in #9482 (comment)