pub fn workspace_optimizations<T, U>(
    nn: *mut NNModule<T, U>,
    ws: *mut Workspace,
    level: i32
)