nbdev-pytorch 0.0.2016__py3-none-any.whl → 0.0.2018__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of nbdev-pytorch might be problematic. Click here for more details.

Files changed (8) hide show
  1. nbdev_pytorch/_modidx.py +137 -69
  2. {nbdev_pytorch-0.0.2016.dist-info → nbdev_pytorch-0.0.2018.dist-info}/METADATA +1 -1
  3. nbdev_pytorch-0.0.2018.dist-info/RECORD +9 -0
  4. nbdev_pytorch-0.0.2016.dist-info/RECORD +0 -9
  5. {nbdev_pytorch-0.0.2016.dist-info → nbdev_pytorch-0.0.2018.dist-info}/WHEEL +0 -0
  6. {nbdev_pytorch-0.0.2016.dist-info → nbdev_pytorch-0.0.2018.dist-info}/entry_points.txt +0 -0
  7. {nbdev_pytorch-0.0.2016.dist-info → nbdev_pytorch-0.0.2018.dist-info}/licenses/LICENSE +0 -0
  8. {nbdev_pytorch-0.0.2016.dist-info → nbdev_pytorch-0.0.2018.dist-info}/top_level.txt +0 -0
nbdev_pytorch/_modidx.py CHANGED
@@ -8,7 +8,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
8
8
  'torch.ComplexDoubleStorage': 'https://pytorch.org/docs/stable/storage.html#torch.ComplexDoubleStorage',
9
9
  'torch.ComplexFloatStorage': 'https://pytorch.org/docs/stable/storage.html#torch.ComplexFloatStorage',
10
10
  'torch.DoubleStorage': 'https://pytorch.org/docs/stable/storage.html#torch.DoubleStorage',
11
- 'torch.Event': 'https://pytorch.org/docs/stable/generated/torch.Event.html#torch.Event',
11
+ 'torch.Event': 'https://pytorch.org/docs/stable/generated/torch.mtia.Event.html#torch.mtia.Event',
12
12
  'torch.Event.elapsed_time': 'https://pytorch.org/docs/stable/generated/torch.Event.html#torch.Event.elapsed_time',
13
13
  'torch.Event.query': 'https://pytorch.org/docs/stable/generated/torch.Event.html#torch.Event.query',
14
14
  'torch.Event.record': 'https://pytorch.org/docs/stable/generated/torch.Event.html#torch.Event.record',
@@ -37,7 +37,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
37
37
  'torch.Size.count': 'https://pytorch.org/docs/stable/size.html#torch.Size.count',
38
38
  'torch.Size.index': 'https://pytorch.org/docs/stable/size.html#torch.Size.index',
39
39
  'torch.Size.numel': 'https://pytorch.org/docs/stable/size.html#torch.Size.numel',
40
- 'torch.Stream': 'https://pytorch.org/docs/stable/generated/torch.Stream.html#torch.Stream',
40
+ 'torch.Stream': 'https://pytorch.org/docs/stable/generated/torch.mtia.Stream.html#torch.mtia.Stream',
41
41
  'torch.Stream.query': 'https://pytorch.org/docs/stable/generated/torch.Stream.html#torch.Stream.query',
42
42
  'torch.Stream.record_event': 'https://pytorch.org/docs/stable/generated/torch.Stream.html#torch.Stream.record_event',
43
43
  'torch.Stream.synchronize': 'https://pytorch.org/docs/stable/generated/torch.Stream.html#torch.Stream.synchronize',
@@ -1228,6 +1228,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1228
1228
  'torch.accelerator.current_device_index': 'https://pytorch.org/docs/stable/generated/torch.accelerator.current_device_index.html#torch.accelerator.current_device_index',
1229
1229
  'torch.accelerator.current_stream': 'https://pytorch.org/docs/stable/generated/torch.accelerator.current_stream.html#torch.accelerator.current_stream',
1230
1230
  'torch.accelerator.device_count': 'https://pytorch.org/docs/stable/generated/torch.accelerator.device_count.html#torch.accelerator.device_count',
1231
+ 'torch.accelerator.device_index': 'https://pytorch.org/docs/stable/generated/torch.accelerator.device_index.html#torch.accelerator.device_index',
1231
1232
  'torch.accelerator.is_available': 'https://pytorch.org/docs/stable/generated/torch.accelerator.is_available.html#torch.accelerator.is_available',
1232
1233
  'torch.accelerator.set_device_idx': 'https://pytorch.org/docs/stable/generated/torch.accelerator.set_device_idx.html#torch.accelerator.set_device_idx',
1233
1234
  'torch.accelerator.set_device_index': 'https://pytorch.org/docs/stable/generated/torch.accelerator.set_device_index.html#torch.accelerator.set_device_index',
@@ -1652,6 +1653,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1652
1653
  'torch.ao.quantization.fx.custom_config.StandaloneModuleConfigEntry': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.fx.custom_config.StandaloneModuleConfigEntry.html#torch.ao.quantization.fx.custom_config.StandaloneModuleConfigEntry'},
1653
1654
  'torch.ao.quantization.observer': { 'torch.ao.quantization.observer.AffineQuantizedObserverBase': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.observer.AffineQuantizedObserverBase.html#torch.ao.quantization.observer.AffineQuantizedObserverBase',
1654
1655
  'torch.ao.quantization.observer.AffineQuantizedObserverBase.calculate_qparams': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.observer.AffineQuantizedObserverBase.html#torch.ao.quantization.observer.AffineQuantizedObserverBase.calculate_qparams',
1656
+ 'torch.ao.quantization.observer.AffineQuantizedObserverBase.convert': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.observer.AffineQuantizedObserverBase.html#torch.ao.quantization.observer.AffineQuantizedObserverBase.convert',
1655
1657
  'torch.ao.quantization.observer.AffineQuantizedObserverBase.forward': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.observer.AffineQuantizedObserverBase.html#torch.ao.quantization.observer.AffineQuantizedObserverBase.forward',
1656
1658
  'torch.ao.quantization.observer.AffineQuantizedObserverBase.with_args': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.observer.AffineQuantizedObserverBase.html#torch.ao.quantization.observer.AffineQuantizedObserverBase.with_args',
1657
1659
  'torch.ao.quantization.observer.Granularity': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.observer.Granularity.html#torch.ao.quantization.observer.Granularity',
@@ -1685,12 +1687,14 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1685
1687
  'torch.ao.quantization.pt2e': { 'torch.ao.quantization.pt2e.duplicate_dq_pass': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.duplicate_dq_pass',
1686
1688
  'torch.ao.quantization.pt2e.export_utils': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.export_utils',
1687
1689
  'torch.ao.quantization.pt2e.graph_utils': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.graph_utils',
1690
+ 'torch.ao.quantization.pt2e.lowering': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.lowering',
1688
1691
  'torch.ao.quantization.pt2e.port_metadata_pass': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.port_metadata_pass',
1689
1692
  'torch.ao.quantization.pt2e.prepare': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.prepare',
1690
1693
  'torch.ao.quantization.pt2e.qat_utils': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.qat_utils',
1691
1694
  'torch.ao.quantization.pt2e.representation': 'https://pytorch.org/docs/stable/quantization-support.html#module-torch.ao.quantization.pt2e.representation',
1692
1695
  'torch.ao.quantization.pt2e.utils': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.utils'},
1693
1696
  'torch.ao.quantization.pt2e.export_utils': { 'torch.ao.quantization.pt2e.export_utils.model_is_exported': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.pt2e.export_utils.model_is_exported.html#torch.ao.quantization.pt2e.export_utils.model_is_exported'},
1697
+ 'torch.ao.quantization.pt2e.lowering': { 'torch.ao.quantization.pt2e.lowering.lower_pt2e_quantized_to_x86': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.pt2e.lowering.lower_pt2e_quantized_to_x86.html#torch.ao.quantization.pt2e.lowering.lower_pt2e_quantized_to_x86'},
1694
1698
  'torch.ao.quantization.pt2e.representation': { 'torch.ao.quantization.pt2e.representation.rewrite': 'https://pytorch.org/docs/stable/quantization.html#module-torch.ao.quantization.pt2e.representation.rewrite'},
1695
1699
  'torch.ao.quantization.qconfig': { 'torch.ao.quantization.qconfig.QConfig': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.qconfig.QConfig.html#torch.ao.quantization.qconfig.QConfig'},
1696
1700
  'torch.ao.quantization.qconfig_mapping': { 'torch.ao.quantization.qconfig_mapping.QConfigMapping': 'https://pytorch.org/docs/stable/generated/torch.ao.quantization.qconfig_mapping.QConfigMapping.html#torch.ao.quantization.qconfig_mapping.QConfigMapping',
@@ -1927,9 +1931,15 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1927
1931
  'torch.compiler.is_compiling': 'https://pytorch.org/docs/stable/generated/torch.compiler.is_compiling.html#torch.compiler.is_compiling',
1928
1932
  'torch.compiler.is_dynamo_compiling': 'https://pytorch.org/docs/stable/generated/torch.compiler.is_dynamo_compiling.html#torch.compiler.is_dynamo_compiling',
1929
1933
  'torch.compiler.is_exporting': 'https://pytorch.org/docs/stable/generated/torch.compiler.is_exporting.html#torch.compiler.is_exporting',
1934
+ 'torch.compiler.keep_tensor_guards_unsafe': 'https://pytorch.org/docs/stable/generated/torch.compiler.keep_tensor_guards_unsafe.html#torch.compiler.keep_tensor_guards_unsafe',
1930
1935
  'torch.compiler.list_backends': 'https://pytorch.org/docs/stable/generated/torch.compiler.list_backends.html#torch.compiler.list_backends',
1936
+ 'torch.compiler.nested_compile_region': 'https://pytorch.org/docs/stable/generated/torch.compiler.nested_compile_region.html#torch.compiler.nested_compile_region',
1931
1937
  'torch.compiler.reset': 'https://pytorch.org/docs/stable/generated/torch.compiler.reset.html#torch.compiler.reset',
1938
+ 'torch.compiler.set_enable_guard_collectives': 'https://pytorch.org/docs/stable/generated/torch.compiler.set_enable_guard_collectives.html#torch.compiler.set_enable_guard_collectives',
1932
1939
  'torch.compiler.set_stance': 'https://pytorch.org/docs/stable/generated/torch.compiler.set_stance.html#torch.compiler.set_stance',
1940
+ 'torch.compiler.skip_guard_on_all_nn_modules_unsafe': 'https://pytorch.org/docs/stable/generated/torch.compiler.skip_guard_on_all_nn_modules_unsafe.html#torch.compiler.skip_guard_on_all_nn_modules_unsafe',
1941
+ 'torch.compiler.skip_guard_on_globals_unsafe': 'https://pytorch.org/docs/stable/generated/torch.compiler.skip_guard_on_globals_unsafe.html#torch.compiler.skip_guard_on_globals_unsafe',
1942
+ 'torch.compiler.skip_guard_on_inbuilt_nn_modules_unsafe': 'https://pytorch.org/docs/stable/generated/torch.compiler.skip_guard_on_inbuilt_nn_modules_unsafe.html#torch.compiler.skip_guard_on_inbuilt_nn_modules_unsafe',
1933
1943
  'torch.compiler.substitute_in_graph': 'https://pytorch.org/docs/stable/generated/torch.compiler.substitute_in_graph.html#torch.compiler.substitute_in_graph'},
1934
1944
  'torch.cpu': { 'torch.cpu.Stream': 'https://pytorch.org/docs/stable/generated/torch.cpu.Stream.html#torch.cpu.Stream',
1935
1945
  'torch.cpu.StreamContext': 'https://pytorch.org/docs/stable/generated/torch.cpu.StreamContext.html#torch.cpu.StreamContext',
@@ -1952,10 +1962,11 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1952
1962
  'torch.cuda.CUDAGraph.capture_end': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.capture_end',
1953
1963
  'torch.cuda.CUDAGraph.debug_dump': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.debug_dump',
1954
1964
  'torch.cuda.CUDAGraph.enable_debug_mode': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.enable_debug_mode',
1965
+ 'torch.cuda.CUDAGraph.instantiate': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.instantiate',
1955
1966
  'torch.cuda.CUDAGraph.pool': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.pool',
1967
+ 'torch.cuda.CUDAGraph.raw_cuda_graph': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.raw_cuda_graph',
1956
1968
  'torch.cuda.CUDAGraph.replay': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.replay',
1957
1969
  'torch.cuda.CUDAGraph.reset': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAGraph.html#torch.cuda.CUDAGraph.reset',
1958
- 'torch.cuda.CUDAPluggableAllocator': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAPluggableAllocator.html#torch.cuda.CUDAPluggableAllocator',
1959
1970
  'torch.cuda.Event': 'https://pytorch.org/docs/stable/generated/torch.cuda.Event.html#torch.cuda.Event',
1960
1971
  'torch.cuda.Event.elapsed_time': 'https://pytorch.org/docs/stable/generated/torch.cuda.Event.html#torch.cuda.Event.elapsed_time',
1961
1972
  'torch.cuda.Event.from_ipc_handle': 'https://pytorch.org/docs/stable/generated/torch.cuda.Event.html#torch.cuda.Event.from_ipc_handle',
@@ -1970,11 +1981,6 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1970
1981
  'torch.cuda.ExternalStream.synchronize': 'https://pytorch.org/docs/stable/generated/torch.cuda.ExternalStream.html#torch.cuda.ExternalStream.synchronize',
1971
1982
  'torch.cuda.ExternalStream.wait_event': 'https://pytorch.org/docs/stable/generated/torch.cuda.ExternalStream.html#torch.cuda.ExternalStream.wait_event',
1972
1983
  'torch.cuda.ExternalStream.wait_stream': 'https://pytorch.org/docs/stable/generated/torch.cuda.ExternalStream.html#torch.cuda.ExternalStream.wait_stream',
1973
- 'torch.cuda.MemPool': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPool.html#torch.cuda.MemPool',
1974
- 'torch.cuda.MemPool.snapshot': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPool.html#torch.cuda.MemPool.snapshot',
1975
- 'torch.cuda.MemPool.use_count': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPool.html#torch.cuda.MemPool.use_count',
1976
- 'torch.cuda.MemPoolContext': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPoolContext.html#torch.cuda.MemPoolContext',
1977
- 'torch.cuda.MemPoolContext.active_pool': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPoolContext.html#torch.cuda.MemPoolContext.active_pool',
1978
1984
  'torch.cuda.Stream': 'https://pytorch.org/docs/stable/generated/torch.cuda.Stream.html#torch.cuda.Stream',
1979
1985
  'torch.cuda.Stream.query': 'https://pytorch.org/docs/stable/generated/torch.cuda.Stream.html#torch.cuda.Stream.query',
1980
1986
  'torch.cuda.Stream.record_event': 'https://pytorch.org/docs/stable/generated/torch.cuda.Stream.html#torch.cuda.Stream.record_event',
@@ -1984,10 +1990,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1984
1990
  'torch.cuda.StreamContext': 'https://pytorch.org/docs/stable/generated/torch.cuda.StreamContext.html#torch.cuda.StreamContext',
1985
1991
  'torch.cuda._sanitizer': 'https://pytorch.org/docs/stable/cuda._sanitizer.html#module-torch.cuda._sanitizer',
1986
1992
  'torch.cuda.amp': 'https://pytorch.org/docs/stable/amp.html#module-torch.cuda.amp',
1987
- 'torch.cuda.caching_allocator_alloc': 'https://pytorch.org/docs/stable/generated/torch.cuda.caching_allocator_alloc.html#torch.cuda.caching_allocator_alloc',
1988
- 'torch.cuda.caching_allocator_delete': 'https://pytorch.org/docs/stable/generated/torch.cuda.caching_allocator_delete.html#torch.cuda.caching_allocator_delete',
1989
1993
  'torch.cuda.can_device_access_peer': 'https://pytorch.org/docs/stable/generated/torch.cuda.can_device_access_peer.html#torch.cuda.can_device_access_peer',
1990
- 'torch.cuda.change_current_allocator': 'https://pytorch.org/docs/stable/generated/torch.cuda.change_current_allocator.html#torch.cuda.change_current_allocator',
1991
1994
  'torch.cuda.clock_rate': 'https://pytorch.org/docs/stable/generated/torch.cuda.clock_rate.html#torch.cuda.clock_rate',
1992
1995
  'torch.cuda.comm': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.comm',
1993
1996
  'torch.cuda.cudart': 'https://pytorch.org/docs/stable/generated/torch.cuda.cudart.html#torch.cuda.cudart',
@@ -1999,16 +2002,13 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
1999
2002
  'torch.cuda.device_count': 'https://pytorch.org/docs/stable/generated/torch.cuda.device_count.html#torch.cuda.device_count',
2000
2003
  'torch.cuda.device_memory_used': 'https://pytorch.org/docs/stable/generated/torch.cuda.device_memory_used.html#torch.cuda.device_memory_used',
2001
2004
  'torch.cuda.device_of': 'https://pytorch.org/docs/stable/generated/torch.cuda.device_of.html#torch.cuda.device_of',
2002
- 'torch.cuda.empty_cache': 'https://pytorch.org/docs/stable/generated/torch.cuda.empty_cache.html#torch.cuda.empty_cache',
2003
2005
  'torch.cuda.error': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.error',
2004
2006
  'torch.cuda.gds': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.gds',
2005
- 'torch.cuda.get_allocator_backend': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_allocator_backend.html#torch.cuda.get_allocator_backend',
2006
2007
  'torch.cuda.get_arch_list': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_arch_list.html#torch.cuda.get_arch_list',
2007
2008
  'torch.cuda.get_device_capability': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_device_capability.html#torch.cuda.get_device_capability',
2008
2009
  'torch.cuda.get_device_name': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_device_name.html#torch.cuda.get_device_name',
2009
2010
  'torch.cuda.get_device_properties': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_device_properties.html#torch.cuda.get_device_properties',
2010
2011
  'torch.cuda.get_gencode_flags': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_gencode_flags.html#torch.cuda.get_gencode_flags',
2011
- 'torch.cuda.get_per_process_memory_fraction': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_per_process_memory_fraction.html#torch.cuda.get_per_process_memory_fraction',
2012
2012
  'torch.cuda.get_rng_state': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_rng_state.html#torch.cuda.get_rng_state',
2013
2013
  'torch.cuda.get_rng_state_all': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_rng_state_all.html#torch.cuda.get_rng_state_all',
2014
2014
  'torch.cuda.get_stream_from_external': 'https://pytorch.org/docs/stable/generated/torch.cuda.get_stream_from_external.html#torch.cuda.get_stream_from_external',
@@ -2016,7 +2016,6 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2016
2016
  'torch.cuda.graph': 'https://pytorch.org/docs/stable/generated/torch.cuda.graph.html#torch.cuda.graph',
2017
2017
  'torch.cuda.graph_pool_handle': 'https://pytorch.org/docs/stable/generated/torch.cuda.graph_pool_handle.html#torch.cuda.graph_pool_handle',
2018
2018
  'torch.cuda.graphs': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.graphs',
2019
- 'torch.cuda.host_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.host_memory_stats.html#torch.cuda.host_memory_stats',
2020
2019
  'torch.cuda.init': 'https://pytorch.org/docs/stable/generated/torch.cuda.init.html#torch.cuda.init',
2021
2020
  'torch.cuda.initial_seed': 'https://pytorch.org/docs/stable/generated/torch.cuda.initial_seed.html#torch.cuda.initial_seed',
2022
2021
  'torch.cuda.ipc_collect': 'https://pytorch.org/docs/stable/generated/torch.cuda.ipc_collect.html#torch.cuda.ipc_collect',
@@ -2025,35 +2024,19 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2025
2024
  'torch.cuda.is_initialized': 'https://pytorch.org/docs/stable/generated/torch.cuda.is_initialized.html#torch.cuda.is_initialized',
2026
2025
  'torch.cuda.is_tf32_supported': 'https://pytorch.org/docs/stable/generated/torch.cuda.is_tf32_supported.html#torch.cuda.is_tf32_supported',
2027
2026
  'torch.cuda.jiterator': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.jiterator',
2028
- 'torch.cuda.list_gpu_processes': 'https://pytorch.org/docs/stable/generated/torch.cuda.list_gpu_processes.html#torch.cuda.list_gpu_processes',
2029
2027
  'torch.cuda.make_graphed_callables': 'https://pytorch.org/docs/stable/generated/torch.cuda.make_graphed_callables.html#torch.cuda.make_graphed_callables',
2030
2028
  'torch.cuda.manual_seed': 'https://pytorch.org/docs/stable/generated/torch.cuda.manual_seed.html#torch.cuda.manual_seed',
2031
2029
  'torch.cuda.manual_seed_all': 'https://pytorch.org/docs/stable/generated/torch.cuda.manual_seed_all.html#torch.cuda.manual_seed_all',
2032
- 'torch.cuda.max_memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.cuda.max_memory_allocated.html#torch.cuda.max_memory_allocated',
2033
- 'torch.cuda.max_memory_cached': 'https://pytorch.org/docs/stable/generated/torch.cuda.max_memory_cached.html#torch.cuda.max_memory_cached',
2034
- 'torch.cuda.max_memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.cuda.max_memory_reserved.html#torch.cuda.max_memory_reserved',
2035
- 'torch.cuda.mem_get_info': 'https://pytorch.org/docs/stable/generated/torch.cuda.mem_get_info.html#torch.cuda.mem_get_info',
2036
2030
  'torch.cuda.memory': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.memory',
2037
- 'torch.cuda.memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_allocated.html#torch.cuda.memory_allocated',
2038
- 'torch.cuda.memory_cached': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_cached.html#torch.cuda.memory_cached',
2039
- 'torch.cuda.memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_reserved.html#torch.cuda.memory_reserved',
2040
- 'torch.cuda.memory_snapshot': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_snapshot.html#torch.cuda.memory_snapshot',
2041
- 'torch.cuda.memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_stats.html#torch.cuda.memory_stats',
2042
- 'torch.cuda.memory_summary': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_summary.html#torch.cuda.memory_summary',
2043
2031
  'torch.cuda.memory_usage': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory_usage.html#torch.cuda.memory_usage',
2044
2032
  'torch.cuda.nccl': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.nccl',
2045
2033
  'torch.cuda.nvtx': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.nvtx',
2046
2034
  'torch.cuda.power_draw': 'https://pytorch.org/docs/stable/generated/torch.cuda.power_draw.html#torch.cuda.power_draw',
2047
2035
  'torch.cuda.profiler': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.profiler',
2048
2036
  'torch.cuda.random': 'https://pytorch.org/docs/stable/cuda.html#module-torch.cuda.random',
2049
- 'torch.cuda.reset_max_memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.cuda.reset_max_memory_allocated.html#torch.cuda.reset_max_memory_allocated',
2050
- 'torch.cuda.reset_max_memory_cached': 'https://pytorch.org/docs/stable/generated/torch.cuda.reset_max_memory_cached.html#torch.cuda.reset_max_memory_cached',
2051
- 'torch.cuda.reset_peak_host_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.reset_peak_host_memory_stats.html#torch.cuda.reset_peak_host_memory_stats',
2052
- 'torch.cuda.reset_peak_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.reset_peak_memory_stats.html#torch.cuda.reset_peak_memory_stats',
2053
2037
  'torch.cuda.seed': 'https://pytorch.org/docs/stable/generated/torch.cuda.seed.html#torch.cuda.seed',
2054
2038
  'torch.cuda.seed_all': 'https://pytorch.org/docs/stable/generated/torch.cuda.seed_all.html#torch.cuda.seed_all',
2055
2039
  'torch.cuda.set_device': 'https://pytorch.org/docs/stable/generated/torch.cuda.set_device.html#torch.cuda.set_device',
2056
- 'torch.cuda.set_per_process_memory_fraction': 'https://pytorch.org/docs/stable/generated/torch.cuda.set_per_process_memory_fraction.html#torch.cuda.set_per_process_memory_fraction',
2057
2040
  'torch.cuda.set_rng_state': 'https://pytorch.org/docs/stable/generated/torch.cuda.set_rng_state.html#torch.cuda.set_rng_state',
2058
2041
  'torch.cuda.set_rng_state_all': 'https://pytorch.org/docs/stable/generated/torch.cuda.set_rng_state_all.html#torch.cuda.set_rng_state_all',
2059
2042
  'torch.cuda.set_stream': 'https://pytorch.org/docs/stable/generated/torch.cuda.set_stream.html#torch.cuda.set_stream',
@@ -2080,6 +2063,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2080
2063
  'torch.cuda.comm.broadcast_coalesced': 'https://pytorch.org/docs/stable/generated/torch.cuda.comm.broadcast_coalesced.html#torch.cuda.comm.broadcast_coalesced',
2081
2064
  'torch.cuda.comm.gather': 'https://pytorch.org/docs/stable/generated/torch.cuda.comm.gather.html#torch.cuda.comm.gather',
2082
2065
  'torch.cuda.comm.reduce_add': 'https://pytorch.org/docs/stable/generated/torch.cuda.comm.reduce_add.html#torch.cuda.comm.reduce_add',
2066
+ 'torch.cuda.comm.reduce_add_coalesced': 'https://pytorch.org/docs/stable/generated/torch.cuda.comm.reduce_add_coalesced.html#torch.cuda.comm.reduce_add_coalesced',
2083
2067
  'torch.cuda.comm.scatter': 'https://pytorch.org/docs/stable/generated/torch.cuda.comm.scatter.html#torch.cuda.comm.scatter'},
2084
2068
  'torch.cuda.gds': { 'torch.cuda.gds.GdsFile': 'https://pytorch.org/docs/stable/generated/torch.cuda.gds.GdsFile.html#torch.cuda.gds.GdsFile',
2085
2069
  'torch.cuda.gds.GdsFile.deregister_handle': 'https://pytorch.org/docs/stable/generated/torch.cuda.gds.GdsFile.html#torch.cuda.gds.GdsFile.deregister_handle',
@@ -2092,13 +2076,41 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2092
2076
  'torch.cuda.graphs.graph': 'https://pytorch.org/docs/stable/generated/torch.cuda.graph.html#torch.cuda.graph'},
2093
2077
  'torch.cuda.jiterator': { 'torch.cuda.jiterator._create_jit_fn': 'https://pytorch.org/docs/stable/generated/torch.cuda.jiterator._create_jit_fn.html#torch.cuda.jiterator._create_jit_fn',
2094
2078
  'torch.cuda.jiterator._create_multi_output_jit_fn': 'https://pytorch.org/docs/stable/generated/torch.cuda.jiterator._create_multi_output_jit_fn.html#torch.cuda.jiterator._create_multi_output_jit_fn'},
2095
- 'torch.cuda.memory': { 'torch.cuda.memory.CUDAPluggableAllocator': 'https://pytorch.org/docs/stable/generated/torch.cuda.CUDAPluggableAllocator.html#torch.cuda.CUDAPluggableAllocator',
2096
- 'torch.cuda.memory.MemPool': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPool.html#torch.cuda.MemPool',
2097
- 'torch.cuda.memory.MemPoolContext': 'https://pytorch.org/docs/stable/generated/torch.cuda.MemPoolContext.html#torch.cuda.MemPoolContext',
2079
+ 'torch.cuda.memory': { 'torch.cuda.memory.CUDAPluggableAllocator': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.CUDAPluggableAllocator.html#torch.cuda.memory.CUDAPluggableAllocator',
2080
+ 'torch.cuda.memory.MemPool': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.MemPool.html#torch.cuda.memory.MemPool',
2081
+ 'torch.cuda.memory.MemPool.snapshot': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.MemPool.html#torch.cuda.memory.MemPool.snapshot',
2082
+ 'torch.cuda.memory.MemPool.use_count': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.MemPool.html#torch.cuda.memory.MemPool.use_count',
2098
2083
  'torch.cuda.memory._dump_snapshot': 'https://pytorch.org/docs/stable/torch_cuda_memory.html#torch.cuda.memory._dump_snapshot',
2099
2084
  'torch.cuda.memory._record_memory_history': 'https://pytorch.org/docs/stable/torch_cuda_memory.html#torch.cuda.memory._record_memory_history',
2100
2085
  'torch.cuda.memory._snapshot': 'https://pytorch.org/docs/stable/torch_cuda_memory.html#torch.cuda.memory._snapshot',
2086
+ 'torch.cuda.memory.caching_allocator_alloc': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.caching_allocator_alloc.html#torch.cuda.memory.caching_allocator_alloc',
2087
+ 'torch.cuda.memory.caching_allocator_delete': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.caching_allocator_delete.html#torch.cuda.memory.caching_allocator_delete',
2101
2088
  'torch.cuda.memory.caching_allocator_enable': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.caching_allocator_enable.html#torch.cuda.memory.caching_allocator_enable',
2089
+ 'torch.cuda.memory.change_current_allocator': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.change_current_allocator.html#torch.cuda.memory.change_current_allocator',
2090
+ 'torch.cuda.memory.empty_cache': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.empty_cache.html#torch.cuda.memory.empty_cache',
2091
+ 'torch.cuda.memory.get_allocator_backend': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.get_allocator_backend.html#torch.cuda.memory.get_allocator_backend',
2092
+ 'torch.cuda.memory.get_per_process_memory_fraction': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.get_per_process_memory_fraction.html#torch.cuda.memory.get_per_process_memory_fraction',
2093
+ 'torch.cuda.memory.host_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.host_memory_stats.html#torch.cuda.memory.host_memory_stats',
2094
+ 'torch.cuda.memory.host_memory_stats_as_nested_dict': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.host_memory_stats_as_nested_dict.html#torch.cuda.memory.host_memory_stats_as_nested_dict',
2095
+ 'torch.cuda.memory.list_gpu_processes': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.list_gpu_processes.html#torch.cuda.memory.list_gpu_processes',
2096
+ 'torch.cuda.memory.max_memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.max_memory_allocated.html#torch.cuda.memory.max_memory_allocated',
2097
+ 'torch.cuda.memory.max_memory_cached': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.max_memory_cached.html#torch.cuda.memory.max_memory_cached',
2098
+ 'torch.cuda.memory.max_memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.max_memory_reserved.html#torch.cuda.memory.max_memory_reserved',
2099
+ 'torch.cuda.memory.mem_get_info': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.mem_get_info.html#torch.cuda.memory.mem_get_info',
2100
+ 'torch.cuda.memory.memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_allocated.html#torch.cuda.memory.memory_allocated',
2101
+ 'torch.cuda.memory.memory_cached': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_cached.html#torch.cuda.memory.memory_cached',
2102
+ 'torch.cuda.memory.memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_reserved.html#torch.cuda.memory.memory_reserved',
2103
+ 'torch.cuda.memory.memory_snapshot': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_snapshot.html#torch.cuda.memory.memory_snapshot',
2104
+ 'torch.cuda.memory.memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_stats.html#torch.cuda.memory.memory_stats',
2105
+ 'torch.cuda.memory.memory_stats_as_nested_dict': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_stats_as_nested_dict.html#torch.cuda.memory.memory_stats_as_nested_dict',
2106
+ 'torch.cuda.memory.memory_summary': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.memory_summary.html#torch.cuda.memory.memory_summary',
2107
+ 'torch.cuda.memory.reset_accumulated_host_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.reset_accumulated_host_memory_stats.html#torch.cuda.memory.reset_accumulated_host_memory_stats',
2108
+ 'torch.cuda.memory.reset_accumulated_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.reset_accumulated_memory_stats.html#torch.cuda.memory.reset_accumulated_memory_stats',
2109
+ 'torch.cuda.memory.reset_max_memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.reset_max_memory_allocated.html#torch.cuda.memory.reset_max_memory_allocated',
2110
+ 'torch.cuda.memory.reset_max_memory_cached': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.reset_max_memory_cached.html#torch.cuda.memory.reset_max_memory_cached',
2111
+ 'torch.cuda.memory.reset_peak_host_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.reset_peak_host_memory_stats.html#torch.cuda.memory.reset_peak_host_memory_stats',
2112
+ 'torch.cuda.memory.reset_peak_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.reset_peak_memory_stats.html#torch.cuda.memory.reset_peak_memory_stats',
2113
+ 'torch.cuda.memory.set_per_process_memory_fraction': 'https://pytorch.org/docs/stable/generated/torch.cuda.memory.set_per_process_memory_fraction.html#torch.cuda.memory.set_per_process_memory_fraction',
2102
2114
  'torch.cuda.memory.use_mem_pool': 'https://pytorch.org/docs/stable/cuda.html#torch.cuda.use_mem_pool'},
2103
2115
  'torch.cuda.nvtx': { 'torch.cuda.nvtx.mark': 'https://pytorch.org/docs/stable/generated/torch.cuda.nvtx.mark.html#torch.cuda.nvtx.mark',
2104
2116
  'torch.cuda.nvtx.range': 'https://pytorch.org/docs/stable/generated/torch.cuda.nvtx.range.html#torch.cuda.nvtx.range',
@@ -2148,6 +2160,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2148
2160
  'torch.distributed.Store.add': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.add',
2149
2161
  'torch.distributed.Store.append': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.append',
2150
2162
  'torch.distributed.Store.check': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.check',
2163
+ 'torch.distributed.Store.clone': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.clone',
2151
2164
  'torch.distributed.Store.compare_set': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.compare_set',
2152
2165
  'torch.distributed.Store.delete_key': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.delete_key',
2153
2166
  'torch.distributed.Store.get': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.get',
@@ -2155,6 +2168,9 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2155
2168
  'torch.distributed.Store.multi_get': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.multi_get',
2156
2169
  'torch.distributed.Store.multi_set': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.multi_set',
2157
2170
  'torch.distributed.Store.num_keys': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.num_keys',
2171
+ 'torch.distributed.Store.queue_len': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.queue_len',
2172
+ 'torch.distributed.Store.queue_pop': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.queue_pop',
2173
+ 'torch.distributed.Store.queue_push': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.queue_push',
2158
2174
  'torch.distributed.Store.set': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.set',
2159
2175
  'torch.distributed.Store.set_timeout': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.set_timeout',
2160
2176
  'torch.distributed.Store.wait': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.Store.wait',
@@ -2197,6 +2213,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2197
2213
  'torch.distributed.gather': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.gather',
2198
2214
  'torch.distributed.gather_object': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.gather_object',
2199
2215
  'torch.distributed.get_backend': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.get_backend',
2216
+ 'torch.distributed.get_default_backend_for_device': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.get_default_backend_for_device',
2200
2217
  'torch.distributed.get_global_rank': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.get_global_rank',
2201
2218
  'torch.distributed.get_group_rank': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.get_group_rank',
2202
2219
  'torch.distributed.get_process_group_ranks': 'https://pytorch.org/docs/stable/distributed.html#torch.distributed.get_process_group_ranks',
@@ -2327,6 +2344,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2327
2344
  'torch.distributed.checkpoint.default_planner': 'https://pytorch.org/docs/stable/distributed.html#module-torch.distributed.checkpoint.default_planner',
2328
2345
  'torch.distributed.checkpoint.filesystem': 'https://pytorch.org/docs/stable/distributed.html#module-torch.distributed.checkpoint.filesystem',
2329
2346
  'torch.distributed.checkpoint.format_utils': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#module-torch.distributed.checkpoint.format_utils',
2347
+ 'torch.distributed.checkpoint.hf_storage': 'https://pytorch.org/docs/stable/distributed.html#module-torch.distributed.checkpoint.hf_storage',
2330
2348
  'torch.distributed.checkpoint.logger': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#module-torch.distributed.checkpoint.logger',
2331
2349
  'torch.distributed.checkpoint.logging_handlers': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#module-torch.distributed.checkpoint.logging_handlers',
2332
2350
  'torch.distributed.checkpoint.metadata': 'https://pytorch.org/docs/stable/distributed.html#module-torch.distributed.checkpoint.metadata',
@@ -2357,7 +2375,8 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2357
2375
  'torch.distributed.checkpoint.format_utils.DynamicMetaLoadPlanner.set_up_planner': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.format_utils.DynamicMetaLoadPlanner.set_up_planner',
2358
2376
  'torch.distributed.checkpoint.format_utils.dcp_to_torch_save': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.format_utils.dcp_to_torch_save',
2359
2377
  'torch.distributed.checkpoint.format_utils.torch_save_to_dcp': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.format_utils.torch_save_to_dcp'},
2360
- 'torch.distributed.checkpoint.planner': { 'torch.distributed.checkpoint.planner.LoadPlan': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.LoadPlan',
2378
+ 'torch.distributed.checkpoint.planner': { 'torch.distributed.checkpoint.planner.BytesIOWriteData': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.planner.BytesIOWriteData',
2379
+ 'torch.distributed.checkpoint.planner.LoadPlan': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.LoadPlan',
2361
2380
  'torch.distributed.checkpoint.planner.LoadPlanner': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.LoadPlanner',
2362
2381
  'torch.distributed.checkpoint.planner.ReadItem': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.ReadItem',
2363
2382
  'torch.distributed.checkpoint.planner.SavePlan': 'https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.SavePlan',
@@ -2586,6 +2605,9 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2586
2605
  'torch.distributed.fsdp.FSDPModule': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule',
2587
2606
  'torch.distributed.fsdp.FSDPModule.reshard': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.reshard',
2588
2607
  'torch.distributed.fsdp.FSDPModule.set_all_reduce_hook': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_all_reduce_hook',
2608
+ 'torch.distributed.fsdp.FSDPModule.set_allocate_memory_from_process_group_for_comm': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_allocate_memory_from_process_group_for_comm',
2609
+ 'torch.distributed.fsdp.FSDPModule.set_force_sum_reduction_for_comms': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_force_sum_reduction_for_comms',
2610
+ 'torch.distributed.fsdp.FSDPModule.set_gradient_divide_factor': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_gradient_divide_factor',
2589
2611
  'torch.distributed.fsdp.FSDPModule.set_is_last_backward': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_is_last_backward',
2590
2612
  'torch.distributed.fsdp.FSDPModule.set_modules_to_backward_prefetch': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_modules_to_backward_prefetch',
2591
2613
  'torch.distributed.fsdp.FSDPModule.set_modules_to_forward_prefetch': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_modules_to_forward_prefetch',
@@ -2594,6 +2616,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2594
2616
  'torch.distributed.fsdp.FSDPModule.set_requires_all_reduce': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_requires_all_reduce',
2595
2617
  'torch.distributed.fsdp.FSDPModule.set_requires_gradient_sync': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_requires_gradient_sync',
2596
2618
  'torch.distributed.fsdp.FSDPModule.set_reshard_after_backward': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_reshard_after_backward',
2619
+ 'torch.distributed.fsdp.FSDPModule.set_reshard_after_forward': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_reshard_after_forward',
2597
2620
  'torch.distributed.fsdp.FSDPModule.set_unshard_in_backward': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.set_unshard_in_backward',
2598
2621
  'torch.distributed.fsdp.FSDPModule.unshard': 'https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html#torch.distributed.fsdp.FSDPModule.unshard',
2599
2622
  'torch.distributed.fsdp.FullOptimStateDictConfig': 'https://pytorch.org/docs/stable/fsdp.html#torch.distributed.fsdp.FullOptimStateDictConfig',
@@ -2785,6 +2808,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2785
2808
  'torch.distributed.tensor.experimental.register_sharding': 'https://pytorch.org/docs/stable/distributed.tensor.html#torch.distributed.tensor.experimental.register_sharding'},
2786
2809
  'torch.distributed.tensor.parallel': { 'torch.distributed.tensor.parallel.ColwiseParallel': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.ColwiseParallel',
2787
2810
  'torch.distributed.tensor.parallel.PrepareModuleInput': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleInput',
2811
+ 'torch.distributed.tensor.parallel.PrepareModuleInputOutput': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleInputOutput',
2788
2812
  'torch.distributed.tensor.parallel.PrepareModuleOutput': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleOutput',
2789
2813
  'torch.distributed.tensor.parallel.RowwiseParallel': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.RowwiseParallel',
2790
2814
  'torch.distributed.tensor.parallel.SequenceParallel': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.SequenceParallel',
@@ -2798,6 +2822,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2798
2822
  'torch.distributed.tensor.parallel.style': 'https://pytorch.org/docs/stable/distributed.html#module-torch.distributed.tensor.parallel.style'},
2799
2823
  'torch.distributed.tensor.parallel.style': { 'torch.distributed.tensor.parallel.style.ColwiseParallel': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.ColwiseParallel',
2800
2824
  'torch.distributed.tensor.parallel.style.PrepareModuleInput': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleInput',
2825
+ 'torch.distributed.tensor.parallel.style.PrepareModuleInputOutput': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleInputOutput',
2801
2826
  'torch.distributed.tensor.parallel.style.PrepareModuleOutput': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleOutput',
2802
2827
  'torch.distributed.tensor.parallel.style.RowwiseParallel': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.RowwiseParallel',
2803
2828
  'torch.distributed.tensor.parallel.style.SequenceParallel': 'https://pytorch.org/docs/stable/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.SequenceParallel'},
@@ -2823,6 +2848,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2823
2848
  'torch.distributions.exponential': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.exponential',
2824
2849
  'torch.distributions.fishersnedecor': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.fishersnedecor',
2825
2850
  'torch.distributions.gamma': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.gamma',
2851
+ 'torch.distributions.generalized_pareto': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.generalized_pareto',
2826
2852
  'torch.distributions.geometric': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.geometric',
2827
2853
  'torch.distributions.gumbel': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.gumbel',
2828
2854
  'torch.distributions.half_cauchy': 'https://pytorch.org/docs/stable/distributions.html#module-torch.distributions.half_cauchy',
@@ -2890,6 +2916,8 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2890
2916
  'torch.distributions.constraint_registry.ConstraintRegistry.register': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.constraint_registry.ConstraintRegistry.register'},
2891
2917
  'torch.distributions.constraints': { 'torch.distributions.constraints.Constraint': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.constraints.Constraint',
2892
2918
  'torch.distributions.constraints.Constraint.check': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.constraints.Constraint.check',
2919
+ 'torch.distributions.constraints.MixtureSameFamilyConstraint': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.constraints.MixtureSameFamilyConstraint',
2920
+ 'torch.distributions.constraints.MixtureSameFamilyConstraint.check': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.constraints.MixtureSameFamilyConstraint.check',
2893
2921
  'torch.distributions.constraints.is_dependent': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.constraints.is_dependent'},
2894
2922
  'torch.distributions.continuous_bernoulli': { 'torch.distributions.continuous_bernoulli.ContinuousBernoulli': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.continuous_bernoulli.ContinuousBernoulli',
2895
2923
  'torch.distributions.continuous_bernoulli.ContinuousBernoulli.cdf': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.continuous_bernoulli.ContinuousBernoulli.cdf',
@@ -2935,6 +2963,15 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
2935
2963
  'torch.distributions.gamma.Gamma.expand': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.gamma.Gamma.expand',
2936
2964
  'torch.distributions.gamma.Gamma.log_prob': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.gamma.Gamma.log_prob',
2937
2965
  'torch.distributions.gamma.Gamma.rsample': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.gamma.Gamma.rsample'},
2966
+ 'torch.distributions.generalized_pareto': { 'torch.distributions.generalized_pareto.GeneralizedPareto': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto',
2967
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.cdf': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.cdf',
2968
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.entropy': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.entropy',
2969
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.expand': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.expand',
2970
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.icdf': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.icdf',
2971
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.log_cdf': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.log_cdf',
2972
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.log_prob': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.log_prob',
2973
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.log_survival_function': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.log_survival_function',
2974
+ 'torch.distributions.generalized_pareto.GeneralizedPareto.rsample': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.generalized_pareto.GeneralizedPareto.rsample'},
2938
2975
  'torch.distributions.geometric': { 'torch.distributions.geometric.Geometric': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.geometric.Geometric',
2939
2976
  'torch.distributions.geometric.Geometric.entropy': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.geometric.Geometric.entropy',
2940
2977
  'torch.distributions.geometric.Geometric.expand': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.geometric.Geometric.expand',
@@ -3091,21 +3128,16 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3091
3128
  'torch.distributions.wishart.Wishart.expand': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.wishart.Wishart.expand',
3092
3129
  'torch.distributions.wishart.Wishart.log_prob': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.wishart.Wishart.log_prob',
3093
3130
  'torch.distributions.wishart.Wishart.rsample': 'https://pytorch.org/docs/stable/distributions.html#torch.distributions.wishart.Wishart.rsample'},
3094
- 'torch.export': { 'torch.export.ExportBackwardSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportBackwardSignature',
3095
- 'torch.export.ExportGraphSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportGraphSignature',
3131
+ 'torch.export': { 'torch.export.ExportGraphSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportGraphSignature',
3096
3132
  'torch.export.ExportedProgram': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram',
3097
- 'torch.export.ExportedProgram.buffers': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram.buffers',
3098
3133
  'torch.export.ExportedProgram.module': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram.module',
3099
- 'torch.export.ExportedProgram.named_buffers': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram.named_buffers',
3100
- 'torch.export.ExportedProgram.named_parameters': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram.named_parameters',
3101
- 'torch.export.ExportedProgram.parameters': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram.parameters',
3102
3134
  'torch.export.ExportedProgram.run_decompositions': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportedProgram.run_decompositions',
3103
3135
  'torch.export.ModuleCallEntry': 'https://pytorch.org/docs/stable/export.html#torch.export.ModuleCallEntry',
3104
3136
  'torch.export.ModuleCallSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.ModuleCallSignature',
3105
3137
  'torch.export.custom_obj': 'https://pytorch.org/docs/stable/export.html#module-torch.export.custom_obj',
3106
3138
  'torch.export.custom_ops': 'https://pytorch.org/docs/stable/export.html#module-torch.export.custom_ops',
3107
3139
  'torch.export.decomp_utils': 'https://pytorch.org/docs/stable/export.html#module-torch.export.decomp_utils',
3108
- 'torch.export.dims': 'https://pytorch.org/docs/stable/export.html#torch.export.dims',
3140
+ 'torch.export.draft_export': 'https://pytorch.org/docs/stable/export.html#torch.export.draft_export',
3109
3141
  'torch.export.dynamic_shapes': 'https://pytorch.org/docs/stable/export.html#module-torch.export.dynamic_shapes',
3110
3142
  'torch.export.experimental': 'https://pytorch.org/docs/stable/export.html#module-torch.export.experimental',
3111
3143
  'torch.export.export': 'https://pytorch.org/docs/stable/export.html#torch.export.export',
@@ -3113,6 +3145,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3113
3145
  'torch.export.graph_signature': 'https://pytorch.org/docs/stable/export.html#module-torch.export.graph_signature',
3114
3146
  'torch.export.load': 'https://pytorch.org/docs/stable/export.html#torch.export.load',
3115
3147
  'torch.export.passes': 'https://pytorch.org/docs/stable/export.html#module-torch.export.passes',
3148
+ 'torch.export.pt2_archive': 'https://pytorch.org/docs/stable/export.html#module-torch.export.pt2_archive',
3116
3149
  'torch.export.register_dataclass': 'https://pytorch.org/docs/stable/export.html#torch.export.register_dataclass',
3117
3150
  'torch.export.save': 'https://pytorch.org/docs/stable/export.html#torch.export.save',
3118
3151
  'torch.export.unflatten': 'https://pytorch.org/docs/stable/export.html#module-torch.export.unflatten'},
@@ -3123,7 +3156,11 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3123
3156
  'torch.export.decomp_utils.CustomDecompTable.materialize': 'https://pytorch.org/docs/stable/export.html#torch.export.decomp_utils.CustomDecompTable.materialize',
3124
3157
  'torch.export.decomp_utils.CustomDecompTable.pop': 'https://pytorch.org/docs/stable/export.html#torch.export.decomp_utils.CustomDecompTable.pop',
3125
3158
  'torch.export.decomp_utils.CustomDecompTable.update': 'https://pytorch.org/docs/stable/export.html#torch.export.decomp_utils.CustomDecompTable.update'},
3126
- 'torch.export.dynamic_shapes': { 'torch.export.dynamic_shapes.Dim': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.Dim',
3159
+ 'torch.export.dynamic_shapes': { 'torch.export.dynamic_shapes.AdditionalInputs': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.AdditionalInputs',
3160
+ 'torch.export.dynamic_shapes.AdditionalInputs.add': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.AdditionalInputs.add',
3161
+ 'torch.export.dynamic_shapes.AdditionalInputs.dynamic_shapes': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.AdditionalInputs.dynamic_shapes',
3162
+ 'torch.export.dynamic_shapes.AdditionalInputs.verify': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.AdditionalInputs.verify',
3163
+ 'torch.export.dynamic_shapes.Dim': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.Dim',
3127
3164
  'torch.export.dynamic_shapes.ShapesCollection': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.ShapesCollection',
3128
3165
  'torch.export.dynamic_shapes.ShapesCollection.dynamic_shapes': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.ShapesCollection.dynamic_shapes',
3129
3166
  'torch.export.dynamic_shapes.refine_dynamic_shapes_from_suggested_fixes': 'https://pytorch.org/docs/stable/export.html#torch.export.dynamic_shapes.refine_dynamic_shapes_from_suggested_fixes'},
@@ -3132,7 +3169,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3132
3169
  'torch.export.exported_program.ModuleCallSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.ModuleCallSignature',
3133
3170
  'torch.export.exported_program.default_decompositions': 'https://pytorch.org/docs/stable/export.html#torch.export.exported_program.default_decompositions'},
3134
3171
  'torch.export.graph_signature': { 'torch.export.graph_signature.CustomObjArgument': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.CustomObjArgument',
3135
- 'torch.export.graph_signature.ExportBackwardSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.ExportBackwardSignature',
3172
+ 'torch.export.graph_signature.ExportBackwardSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.ExportBackwardSignature',
3136
3173
  'torch.export.graph_signature.ExportGraphSignature': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.ExportGraphSignature',
3137
3174
  'torch.export.graph_signature.ExportGraphSignature.get_replace_hook': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.ExportGraphSignature.get_replace_hook',
3138
3175
  'torch.export.graph_signature.ExportGraphSignature.replace_all_uses': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.ExportGraphSignature.replace_all_uses',
@@ -3144,6 +3181,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3144
3181
  'torch.export.graph_signature.SymFloatArgument': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.SymFloatArgument',
3145
3182
  'torch.export.graph_signature.SymIntArgument': 'https://pytorch.org/docs/stable/export.html#torch.export.graph_signature.SymIntArgument'},
3146
3183
  'torch.export.passes': { 'torch.export.passes.move_to_device_pass': 'https://pytorch.org/docs/stable/export.html#torch.export.passes.move_to_device_pass'},
3184
+ 'torch.export.pt2_archive': { 'torch.export.pt2_archive.constants': 'https://pytorch.org/docs/stable/export.html#module-torch.export.pt2_archive.constants'},
3147
3185
  'torch.export.unflatten': { 'torch.export.unflatten.FlatArgsAdapter': 'https://pytorch.org/docs/stable/export.html#torch.export.unflatten.FlatArgsAdapter',
3148
3186
  'torch.export.unflatten.FlatArgsAdapter.adapt': 'https://pytorch.org/docs/stable/export.html#torch.export.unflatten.FlatArgsAdapter.adapt',
3149
3187
  'torch.export.unflatten.InterpreterModule': 'https://pytorch.org/docs/stable/export.html#torch.export.unflatten.InterpreterModule',
@@ -3370,8 +3408,8 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3370
3408
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.create_unbacked_symint': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.create_unbacked_symint',
3371
3409
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.create_unspecified_symbol': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.create_unspecified_symbol',
3372
3410
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.create_unspecified_symint_and_symbol': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.create_unspecified_symint_and_symbol',
3373
- 'torch.fx.experimental.symbolic_shapes.ShapeEnv.defer_runtime_assert': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.defer_runtime_assert',
3374
3411
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.deserialize_symexpr': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.deserialize_symexpr',
3412
+ 'torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_expr': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_expr',
3375
3413
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_guards_expression': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_guards_expression',
3376
3414
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_guards_for_args': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_guards_for_args',
3377
3415
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_sym_node': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_sym_node',
@@ -3383,8 +3421,10 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3383
3421
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.get_implications': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.get_implications',
3384
3422
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.get_nontrivial_guards': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.get_nontrivial_guards',
3385
3423
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.get_pruned_guards': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.get_pruned_guards',
3424
+ 'torch.fx.experimental.symbolic_shapes.ShapeEnv.guard_or_defer_runtime_assert': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.guard_or_defer_runtime_assert',
3386
3425
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.ignore_fresh_unbacked_symbols': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.ignore_fresh_unbacked_symbols',
3387
3426
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.is_unbacked_symint': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.is_unbacked_symint',
3427
+ 'torch.fx.experimental.symbolic_shapes.ShapeEnv.patch_source_specialization': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.patch_source_specialization',
3388
3428
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.produce_guards': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.produce_guards',
3389
3429
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.produce_guards_expression': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.produce_guards_expression',
3390
3430
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.produce_guards_verbose': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.produce_guards_verbose',
@@ -3394,6 +3434,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3394
3434
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.size_hint': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.size_hint',
3395
3435
  'torch.fx.experimental.symbolic_shapes.ShapeEnv.suppress_guards': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnv.html#torch.fx.experimental.symbolic_shapes.ShapeEnv.suppress_guards',
3396
3436
  'torch.fx.experimental.symbolic_shapes.ShapeEnvSettings': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.ShapeEnvSettings.html#torch.fx.experimental.symbolic_shapes.ShapeEnvSettings',
3437
+ 'torch.fx.experimental.symbolic_shapes.Specialization': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.Specialization.html#torch.fx.experimental.symbolic_shapes.Specialization',
3397
3438
  'torch.fx.experimental.symbolic_shapes.StatefulSymbolicContext': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.StatefulSymbolicContext.html#torch.fx.experimental.symbolic_shapes.StatefulSymbolicContext',
3398
3439
  'torch.fx.experimental.symbolic_shapes.StatelessSymbolicContext': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.StatelessSymbolicContext.html#torch.fx.experimental.symbolic_shapes.StatelessSymbolicContext',
3399
3440
  'torch.fx.experimental.symbolic_shapes.StrictMinMaxConstraint': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.StrictMinMaxConstraint.html#torch.fx.experimental.symbolic_shapes.StrictMinMaxConstraint',
@@ -3405,11 +3446,12 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3405
3446
  'torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings.html#torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings',
3406
3447
  'torch.fx.experimental.symbolic_shapes.constrain_range': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.constrain_range.html#torch.fx.experimental.symbolic_shapes.constrain_range',
3407
3448
  'torch.fx.experimental.symbolic_shapes.constrain_unify': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.constrain_unify.html#torch.fx.experimental.symbolic_shapes.constrain_unify',
3408
- 'torch.fx.experimental.symbolic_shapes.definitely_false': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.definitely_false.html#torch.fx.experimental.symbolic_shapes.definitely_false',
3409
- 'torch.fx.experimental.symbolic_shapes.definitely_true': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.definitely_true.html#torch.fx.experimental.symbolic_shapes.definitely_true',
3449
+ 'torch.fx.experimental.symbolic_shapes.guard_or_false': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.guard_or_false.html#torch.fx.experimental.symbolic_shapes.guard_or_false',
3450
+ 'torch.fx.experimental.symbolic_shapes.guard_or_true': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.guard_or_true.html#torch.fx.experimental.symbolic_shapes.guard_or_true',
3410
3451
  'torch.fx.experimental.symbolic_shapes.guard_size_oblivious': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.guard_size_oblivious.html#torch.fx.experimental.symbolic_shapes.guard_size_oblivious',
3411
3452
  'torch.fx.experimental.symbolic_shapes.has_free_symbols': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.has_free_symbols.html#torch.fx.experimental.symbolic_shapes.has_free_symbols',
3412
3453
  'torch.fx.experimental.symbolic_shapes.has_free_unbacked_symbols': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.has_free_unbacked_symbols.html#torch.fx.experimental.symbolic_shapes.has_free_unbacked_symbols',
3454
+ 'torch.fx.experimental.symbolic_shapes.has_static_value': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.has_static_value.html#torch.fx.experimental.symbolic_shapes.has_static_value',
3413
3455
  'torch.fx.experimental.symbolic_shapes.hint_int': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.hint_int.html#torch.fx.experimental.symbolic_shapes.hint_int',
3414
3456
  'torch.fx.experimental.symbolic_shapes.is_accessor_node': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.is_accessor_node.html#torch.fx.experimental.symbolic_shapes.is_accessor_node',
3415
3457
  'torch.fx.experimental.symbolic_shapes.is_concrete_bool': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.is_concrete_bool.html#torch.fx.experimental.symbolic_shapes.is_concrete_bool',
@@ -3418,8 +3460,11 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3418
3460
  'torch.fx.experimental.symbolic_shapes.lru_cache': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.lru_cache.html#torch.fx.experimental.symbolic_shapes.lru_cache',
3419
3461
  'torch.fx.experimental.symbolic_shapes.rebind_unbacked': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.rebind_unbacked.html#torch.fx.experimental.symbolic_shapes.rebind_unbacked',
3420
3462
  'torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings.html#torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings',
3463
+ 'torch.fx.experimental.symbolic_shapes.statically_known_false': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.statically_known_false.html#torch.fx.experimental.symbolic_shapes.statically_known_false',
3421
3464
  'torch.fx.experimental.symbolic_shapes.statically_known_true': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.statically_known_true.html#torch.fx.experimental.symbolic_shapes.statically_known_true',
3422
- 'torch.fx.experimental.symbolic_shapes.sym_eq': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.sym_eq.html#torch.fx.experimental.symbolic_shapes.sym_eq'},
3465
+ 'torch.fx.experimental.symbolic_shapes.sym_and': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.sym_and.html#torch.fx.experimental.symbolic_shapes.sym_and',
3466
+ 'torch.fx.experimental.symbolic_shapes.sym_eq': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.sym_eq.html#torch.fx.experimental.symbolic_shapes.sym_eq',
3467
+ 'torch.fx.experimental.symbolic_shapes.sym_or': 'https://pytorch.org/docs/stable/generated/torch.fx.experimental.symbolic_shapes.sym_or.html#torch.fx.experimental.symbolic_shapes.sym_or'},
3423
3468
  'torch.fx.experimental.unification': { 'torch.fx.experimental.unification.core': 'https://pytorch.org/docs/stable/fx.html#module-torch.fx.experimental.unification.core',
3424
3469
  'torch.fx.experimental.unification.dispatch': 'https://pytorch.org/docs/stable/fx.html#module-torch.fx.experimental.unification.dispatch',
3425
3470
  'torch.fx.experimental.unification.match': 'https://pytorch.org/docs/stable/fx.html#module-torch.fx.experimental.unification.match',
@@ -3691,6 +3736,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3691
3736
  'torch.mtia.Stream.wait_event': 'https://pytorch.org/docs/stable/generated/torch.mtia.Stream.html#torch.mtia.Stream.wait_event',
3692
3737
  'torch.mtia.Stream.wait_stream': 'https://pytorch.org/docs/stable/generated/torch.mtia.Stream.html#torch.mtia.Stream.wait_stream',
3693
3738
  'torch.mtia.StreamContext': 'https://pytorch.org/docs/stable/generated/torch.mtia.StreamContext.html#torch.mtia.StreamContext',
3739
+ 'torch.mtia.attach_out_of_memory_observer': 'https://pytorch.org/docs/stable/generated/torch.mtia.attach_out_of_memory_observer.html#torch.mtia.attach_out_of_memory_observer',
3694
3740
  'torch.mtia.current_device': 'https://pytorch.org/docs/stable/generated/torch.mtia.current_device.html#torch.mtia.current_device',
3695
3741
  'torch.mtia.current_stream': 'https://pytorch.org/docs/stable/generated/torch.mtia.current_stream.html#torch.mtia.current_stream',
3696
3742
  'torch.mtia.default_stream': 'https://pytorch.org/docs/stable/generated/torch.mtia.default_stream.html#torch.mtia.default_stream',
@@ -3761,6 +3807,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3761
3807
  'torch.nn.Conv3d': 'https://pytorch.org/docs/stable/generated/torch.nn.Conv3d.html#torch.nn.Conv3d',
3762
3808
  'torch.nn.ConvTranspose1d': 'https://pytorch.org/docs/stable/generated/torch.nn.ConvTranspose1d.html#torch.nn.ConvTranspose1d',
3763
3809
  'torch.nn.ConvTranspose2d': 'https://pytorch.org/docs/stable/generated/torch.nn.ConvTranspose2d.html#torch.nn.ConvTranspose2d',
3810
+ 'torch.nn.ConvTranspose2d.forward': 'https://pytorch.org/docs/stable/generated/torch.nn.ConvTranspose2d.html#torch.nn.ConvTranspose2d.forward',
3764
3811
  'torch.nn.ConvTranspose3d': 'https://pytorch.org/docs/stable/generated/torch.nn.ConvTranspose3d.html#torch.nn.ConvTranspose3d',
3765
3812
  'torch.nn.CosineEmbeddingLoss': 'https://pytorch.org/docs/stable/generated/torch.nn.CosineEmbeddingLoss.html#torch.nn.CosineEmbeddingLoss',
3766
3813
  'torch.nn.CosineSimilarity': 'https://pytorch.org/docs/stable/generated/torch.nn.CosineSimilarity.html#torch.nn.CosineSimilarity',
@@ -3941,6 +3988,8 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
3941
3988
  'torch.nn.SELU': 'https://pytorch.org/docs/stable/generated/torch.nn.SELU.html#torch.nn.SELU',
3942
3989
  'torch.nn.Sequential': 'https://pytorch.org/docs/stable/generated/torch.nn.Sequential.html#torch.nn.Sequential',
3943
3990
  'torch.nn.Sequential.append': 'https://pytorch.org/docs/stable/generated/torch.nn.Sequential.html#torch.nn.Sequential.append',
3991
+ 'torch.nn.Sequential.extend': 'https://pytorch.org/docs/stable/generated/torch.nn.Sequential.html#torch.nn.Sequential.extend',
3992
+ 'torch.nn.Sequential.insert': 'https://pytorch.org/docs/stable/generated/torch.nn.Sequential.html#torch.nn.Sequential.insert',
3944
3993
  'torch.nn.SiLU': 'https://pytorch.org/docs/stable/generated/torch.nn.SiLU.html#torch.nn.SiLU',
3945
3994
  'torch.nn.Sigmoid': 'https://pytorch.org/docs/stable/generated/torch.nn.Sigmoid.html#torch.nn.Sigmoid',
3946
3995
  'torch.nn.SmoothL1Loss': 'https://pytorch.org/docs/stable/generated/torch.nn.SmoothL1Loss.html#torch.nn.SmoothL1Loss',
@@ -4504,8 +4553,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
4504
4553
  'torch.nn.utils.rnn.unpack_sequence': 'https://pytorch.org/docs/stable/generated/torch.nn.utils.rnn.unpack_sequence.html#torch.nn.utils.rnn.unpack_sequence',
4505
4554
  'torch.nn.utils.rnn.unpad_sequence': 'https://pytorch.org/docs/stable/generated/torch.nn.utils.rnn.unpad_sequence.html#torch.nn.utils.rnn.unpad_sequence'},
4506
4555
  'torch.nn.utils.stateless': { 'torch.nn.utils.stateless.functional_call': 'https://pytorch.org/docs/stable/generated/torch.nn.utils.stateless.functional_call.html#torch.nn.utils.stateless.functional_call'},
4507
- 'torch.onnx': { 'torch.onnx.DiagnosticOptions': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.DiagnosticOptions',
4508
- 'torch.onnx.ExportOptions': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.ExportOptions',
4556
+ 'torch.onnx': { 'torch.onnx.ExportOptions': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.ExportOptions',
4509
4557
  'torch.onnx.JitScalarType': 'https://pytorch.org/docs/stable/generated/torch.onnx.JitScalarType.html#torch.onnx.JitScalarType',
4510
4558
  'torch.onnx.JitScalarType.dtype': 'https://pytorch.org/docs/stable/generated/torch.onnx.JitScalarType.html#torch.onnx.JitScalarType.dtype',
4511
4559
  'torch.onnx.JitScalarType.from_dtype': 'https://pytorch.org/docs/stable/generated/torch.onnx.JitScalarType.html#torch.onnx.JitScalarType.from_dtype',
@@ -4522,19 +4570,15 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
4522
4570
  'torch.onnx.ONNXProgram.optimize': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.ONNXProgram.optimize',
4523
4571
  'torch.onnx.ONNXProgram.release': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.ONNXProgram.release',
4524
4572
  'torch.onnx.ONNXProgram.save': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.ONNXProgram.save',
4525
- 'torch.onnx.ONNXRuntimeOptions': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.ONNXRuntimeOptions',
4526
4573
  'torch.onnx.OnnxExporterError': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.OnnxExporterError',
4527
- 'torch.onnx.OnnxRegistry': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.OnnxRegistry',
4528
- 'torch.onnx.OnnxRegistry.get_op_functions': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.OnnxRegistry.get_op_functions',
4529
- 'torch.onnx.OnnxRegistry.is_registered_op': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.OnnxRegistry.is_registered_op',
4530
- 'torch.onnx.OnnxRegistry.register_op': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.OnnxRegistry.register_op',
4531
4574
  'torch.onnx.dynamo_export': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.dynamo_export',
4532
4575
  'torch.onnx.enable_fake_mode': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.enable_fake_mode',
4533
4576
  'torch.onnx.errors': 'https://pytorch.org/docs/stable/onnx.html#module-torch.onnx.errors',
4534
- 'torch.onnx.export': 'https://pytorch.org/docs/stable/onnx_torchscript.html#torch.onnx.export',
4535
- 'torch.onnx.is_in_onnx_export': 'https://pytorch.org/docs/stable/onnx_torchscript.html#torch.onnx.is_in_onnx_export',
4577
+ 'torch.onnx.export': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.export',
4578
+ 'torch.onnx.is_in_onnx_export': 'https://pytorch.org/docs/stable/onnx_dynamo.html#torch.onnx.is_in_onnx_export',
4536
4579
  'torch.onnx.is_onnxrt_backend_supported': 'https://pytorch.org/docs/stable/onnx_dynamo_onnxruntime_backend.html#torch.onnx.is_onnxrt_backend_supported',
4537
4580
  'torch.onnx.operators': 'https://pytorch.org/docs/stable/onnx.html#module-torch.onnx.operators',
4581
+ 'torch.onnx.ops': 'https://pytorch.org/docs/stable/onnx_ops.html#module-torch.onnx.ops',
4538
4582
  'torch.onnx.register_custom_op_symbolic': 'https://pytorch.org/docs/stable/onnx_torchscript.html#torch.onnx.register_custom_op_symbolic',
4539
4583
  'torch.onnx.select_model_mode_for_export': 'https://pytorch.org/docs/stable/onnx_torchscript.html#torch.onnx.select_model_mode_for_export',
4540
4584
  'torch.onnx.symbolic_caffe2': 'https://pytorch.org/docs/stable/onnx.html#module-torch.onnx.symbolic_caffe2',
@@ -4556,11 +4600,17 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
4556
4600
  'torch.onnx.unregister_custom_op_symbolic': 'https://pytorch.org/docs/stable/onnx_torchscript.html#torch.onnx.unregister_custom_op_symbolic',
4557
4601
  'torch.onnx.utils': 'https://pytorch.org/docs/stable/onnx.html#module-torch.onnx.utils',
4558
4602
  'torch.onnx.verification': 'https://pytorch.org/docs/stable/onnx_verification.html#module-torch.onnx.verification'},
4603
+ 'torch.onnx.ops': { 'torch.onnx.ops.aten_decompositions': 'https://pytorch.org/docs/stable/onnx_ops.html#torch.onnx.ops.aten_decompositions',
4604
+ 'torch.onnx.ops.attention': 'https://pytorch.org/docs/stable/onnx_ops.html#torch.onnx.ops.attention',
4605
+ 'torch.onnx.ops.rotary_embedding': 'https://pytorch.org/docs/stable/onnx_ops.html#torch.onnx.ops.rotary_embedding',
4606
+ 'torch.onnx.ops.symbolic': 'https://pytorch.org/docs/stable/onnx_ops.html#torch.onnx.ops.symbolic',
4607
+ 'torch.onnx.ops.symbolic_multi_out': 'https://pytorch.org/docs/stable/onnx_ops.html#torch.onnx.ops.symbolic_multi_out'},
4559
4608
  'torch.onnx.verification': { 'torch.onnx.verification.GraphInfo': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.GraphInfo',
4560
4609
  'torch.onnx.verification.GraphInfoPrettyPrinter': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.GraphInfoPrettyPrinter',
4561
4610
  'torch.onnx.verification.OnnxBackend': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.OnnxBackend',
4562
4611
  'torch.onnx.verification.OnnxTestCaseRepro': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.OnnxTestCaseRepro',
4563
4612
  'torch.onnx.verification.VerificationInfo': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.VerificationInfo',
4613
+ 'torch.onnx.verification.VerificationInfo.asdict': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.VerificationInfo.asdict',
4564
4614
  'torch.onnx.verification.VerificationInfo.from_tensors': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.VerificationInfo.from_tensors',
4565
4615
  'torch.onnx.verification.VerificationOptions': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.VerificationOptions',
4566
4616
  'torch.onnx.verification.check_export_model_diff': 'https://pytorch.org/docs/stable/onnx_verification.html#torch.onnx.verification.check_export_model_diff',
@@ -4853,6 +4903,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
4853
4903
  'torch.optim.lr_scheduler.ReduceLROnPlateau.get_last_lr': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.ReduceLROnPlateau.html#torch.optim.lr_scheduler.ReduceLROnPlateau.get_last_lr',
4854
4904
  'torch.optim.lr_scheduler.ReduceLROnPlateau.get_lr': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.ReduceLROnPlateau.html#torch.optim.lr_scheduler.ReduceLROnPlateau.get_lr',
4855
4905
  'torch.optim.lr_scheduler.ReduceLROnPlateau.load_state_dict': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.ReduceLROnPlateau.html#torch.optim.lr_scheduler.ReduceLROnPlateau.load_state_dict',
4906
+ 'torch.optim.lr_scheduler.ReduceLROnPlateau.state_dict': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.ReduceLROnPlateau.html#torch.optim.lr_scheduler.ReduceLROnPlateau.state_dict',
4856
4907
  'torch.optim.lr_scheduler.ReduceLROnPlateau.step': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.ReduceLROnPlateau.html#torch.optim.lr_scheduler.ReduceLROnPlateau.step',
4857
4908
  'torch.optim.lr_scheduler.SequentialLR': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.SequentialLR.html#torch.optim.lr_scheduler.SequentialLR',
4858
4909
  'torch.optim.lr_scheduler.SequentialLR.get_last_lr': 'https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.SequentialLR.html#torch.optim.lr_scheduler.SequentialLR.get_last_lr',
@@ -5101,6 +5152,12 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5101
5152
  'torch.special': { 'torch.special.airy_ai': 'https://pytorch.org/docs/stable/special.html#torch.special.airy_ai',
5102
5153
  'torch.special.bessel_j0': 'https://pytorch.org/docs/stable/special.html#torch.special.bessel_j0',
5103
5154
  'torch.special.bessel_j1': 'https://pytorch.org/docs/stable/special.html#torch.special.bessel_j1',
5155
+ 'torch.special.bessel_y0': 'https://pytorch.org/docs/stable/special.html#torch.special.bessel_y0',
5156
+ 'torch.special.bessel_y1': 'https://pytorch.org/docs/stable/special.html#torch.special.bessel_y1',
5157
+ 'torch.special.chebyshev_polynomial_t': 'https://pytorch.org/docs/stable/special.html#torch.special.chebyshev_polynomial_t',
5158
+ 'torch.special.chebyshev_polynomial_u': 'https://pytorch.org/docs/stable/special.html#torch.special.chebyshev_polynomial_u',
5159
+ 'torch.special.chebyshev_polynomial_v': 'https://pytorch.org/docs/stable/special.html#torch.special.chebyshev_polynomial_v',
5160
+ 'torch.special.chebyshev_polynomial_w': 'https://pytorch.org/docs/stable/special.html#torch.special.chebyshev_polynomial_w',
5104
5161
  'torch.special.digamma': 'https://pytorch.org/docs/stable/special.html#torch.special.digamma',
5105
5162
  'torch.special.entr': 'https://pytorch.org/docs/stable/special.html#torch.special.entr',
5106
5163
  'torch.special.erf': 'https://pytorch.org/docs/stable/special.html#torch.special.erf',
@@ -5113,15 +5170,23 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5113
5170
  'torch.special.gammainc': 'https://pytorch.org/docs/stable/special.html#torch.special.gammainc',
5114
5171
  'torch.special.gammaincc': 'https://pytorch.org/docs/stable/special.html#torch.special.gammaincc',
5115
5172
  'torch.special.gammaln': 'https://pytorch.org/docs/stable/special.html#torch.special.gammaln',
5173
+ 'torch.special.hermite_polynomial_h': 'https://pytorch.org/docs/stable/special.html#torch.special.hermite_polynomial_h',
5174
+ 'torch.special.hermite_polynomial_he': 'https://pytorch.org/docs/stable/special.html#torch.special.hermite_polynomial_he',
5116
5175
  'torch.special.i0': 'https://pytorch.org/docs/stable/special.html#torch.special.i0',
5117
5176
  'torch.special.i0e': 'https://pytorch.org/docs/stable/special.html#torch.special.i0e',
5118
5177
  'torch.special.i1': 'https://pytorch.org/docs/stable/special.html#torch.special.i1',
5119
5178
  'torch.special.i1e': 'https://pytorch.org/docs/stable/special.html#torch.special.i1e',
5179
+ 'torch.special.laguerre_polynomial_l': 'https://pytorch.org/docs/stable/special.html#torch.special.laguerre_polynomial_l',
5180
+ 'torch.special.legendre_polynomial_p': 'https://pytorch.org/docs/stable/special.html#torch.special.legendre_polynomial_p',
5120
5181
  'torch.special.log1p': 'https://pytorch.org/docs/stable/special.html#torch.special.log1p',
5121
5182
  'torch.special.log_ndtr': 'https://pytorch.org/docs/stable/special.html#torch.special.log_ndtr',
5122
5183
  'torch.special.log_softmax': 'https://pytorch.org/docs/stable/special.html#torch.special.log_softmax',
5123
5184
  'torch.special.logit': 'https://pytorch.org/docs/stable/special.html#torch.special.logit',
5124
5185
  'torch.special.logsumexp': 'https://pytorch.org/docs/stable/special.html#torch.special.logsumexp',
5186
+ 'torch.special.modified_bessel_i0': 'https://pytorch.org/docs/stable/special.html#torch.special.modified_bessel_i0',
5187
+ 'torch.special.modified_bessel_i1': 'https://pytorch.org/docs/stable/special.html#torch.special.modified_bessel_i1',
5188
+ 'torch.special.modified_bessel_k0': 'https://pytorch.org/docs/stable/special.html#torch.special.modified_bessel_k0',
5189
+ 'torch.special.modified_bessel_k1': 'https://pytorch.org/docs/stable/special.html#torch.special.modified_bessel_k1',
5125
5190
  'torch.special.multigammaln': 'https://pytorch.org/docs/stable/special.html#torch.special.multigammaln',
5126
5191
  'torch.special.ndtr': 'https://pytorch.org/docs/stable/special.html#torch.special.ndtr',
5127
5192
  'torch.special.ndtri': 'https://pytorch.org/docs/stable/special.html#torch.special.ndtri',
@@ -5130,6 +5195,10 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5130
5195
  'torch.special.round': 'https://pytorch.org/docs/stable/special.html#torch.special.round',
5131
5196
  'torch.special.scaled_modified_bessel_k0': 'https://pytorch.org/docs/stable/special.html#torch.special.scaled_modified_bessel_k0',
5132
5197
  'torch.special.scaled_modified_bessel_k1': 'https://pytorch.org/docs/stable/special.html#torch.special.scaled_modified_bessel_k1',
5198
+ 'torch.special.shifted_chebyshev_polynomial_t': 'https://pytorch.org/docs/stable/special.html#torch.special.shifted_chebyshev_polynomial_t',
5199
+ 'torch.special.shifted_chebyshev_polynomial_u': 'https://pytorch.org/docs/stable/special.html#torch.special.shifted_chebyshev_polynomial_u',
5200
+ 'torch.special.shifted_chebyshev_polynomial_v': 'https://pytorch.org/docs/stable/special.html#torch.special.shifted_chebyshev_polynomial_v',
5201
+ 'torch.special.shifted_chebyshev_polynomial_w': 'https://pytorch.org/docs/stable/special.html#torch.special.shifted_chebyshev_polynomial_w',
5133
5202
  'torch.special.sinc': 'https://pytorch.org/docs/stable/special.html#torch.special.sinc',
5134
5203
  'torch.special.softmax': 'https://pytorch.org/docs/stable/special.html#torch.special.softmax',
5135
5204
  'torch.special.spherical_bessel_j0': 'https://pytorch.org/docs/stable/special.html#torch.special.spherical_bessel_j0',
@@ -5201,8 +5270,7 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5201
5270
  'torch.utils.benchmark.examples': 'https://pytorch.org/docs/stable/benchmark_utils.html#module-torch.utils.benchmark.examples',
5202
5271
  'torch.utils.benchmark.op_fuzzers': 'https://pytorch.org/docs/stable/benchmark_utils.html#module-torch.utils.benchmark.op_fuzzers',
5203
5272
  'torch.utils.benchmark.utils': 'https://pytorch.org/docs/stable/benchmark_utils.html#module-torch.utils.benchmark.utils'},
5204
- 'torch.utils.benchmark.examples': { 'torch.utils.benchmark.examples.blas_compare_setup': 'https://pytorch.org/docs/stable/utils.html#module-torch.utils.benchmark.examples.blas_compare_setup',
5205
- 'torch.utils.benchmark.examples.compare': 'https://pytorch.org/docs/stable/utils.html#module-torch.utils.benchmark.examples.compare',
5273
+ 'torch.utils.benchmark.examples': { 'torch.utils.benchmark.examples.compare': 'https://pytorch.org/docs/stable/utils.html#module-torch.utils.benchmark.examples.compare',
5206
5274
  'torch.utils.benchmark.examples.fuzzer': 'https://pytorch.org/docs/stable/utils.html#module-torch.utils.benchmark.examples.fuzzer',
5207
5275
  'torch.utils.benchmark.examples.op_benchmark': 'https://pytorch.org/docs/stable/utils.html#module-torch.utils.benchmark.examples.op_benchmark',
5208
5276
  'torch.utils.benchmark.examples.simple_timeit': 'https://pytorch.org/docs/stable/utils.html#module-torch.utils.benchmark.examples.simple_timeit',
@@ -5363,7 +5431,6 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5363
5431
  'torch.xpu.device': 'https://pytorch.org/docs/stable/generated/torch.xpu.device.html#torch.xpu.device',
5364
5432
  'torch.xpu.device_count': 'https://pytorch.org/docs/stable/generated/torch.xpu.device_count.html#torch.xpu.device_count',
5365
5433
  'torch.xpu.device_of': 'https://pytorch.org/docs/stable/generated/torch.xpu.device_of.html#torch.xpu.device_of',
5366
- 'torch.xpu.empty_cache': 'https://pytorch.org/docs/stable/generated/torch.xpu.empty_cache.html#torch.xpu.empty_cache',
5367
5434
  'torch.xpu.get_arch_list': 'https://pytorch.org/docs/stable/generated/torch.xpu.get_arch_list.html#torch.xpu.get_arch_list',
5368
5435
  'torch.xpu.get_device_capability': 'https://pytorch.org/docs/stable/generated/torch.xpu.get_device_capability.html#torch.xpu.get_device_capability',
5369
5436
  'torch.xpu.get_device_name': 'https://pytorch.org/docs/stable/generated/torch.xpu.get_device_name.html#torch.xpu.get_device_name',
@@ -5378,17 +5445,8 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5378
5445
  'torch.xpu.is_initialized': 'https://pytorch.org/docs/stable/generated/torch.xpu.is_initialized.html#torch.xpu.is_initialized',
5379
5446
  'torch.xpu.manual_seed': 'https://pytorch.org/docs/stable/generated/torch.xpu.manual_seed.html#torch.xpu.manual_seed',
5380
5447
  'torch.xpu.manual_seed_all': 'https://pytorch.org/docs/stable/generated/torch.xpu.manual_seed_all.html#torch.xpu.manual_seed_all',
5381
- 'torch.xpu.max_memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.xpu.max_memory_allocated.html#torch.xpu.max_memory_allocated',
5382
- 'torch.xpu.max_memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.xpu.max_memory_reserved.html#torch.xpu.max_memory_reserved',
5383
- 'torch.xpu.mem_get_info': 'https://pytorch.org/docs/stable/generated/torch.xpu.mem_get_info.html#torch.xpu.mem_get_info',
5384
5448
  'torch.xpu.memory': 'https://pytorch.org/docs/stable/xpu.html#module-torch.xpu.memory',
5385
- 'torch.xpu.memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory_allocated.html#torch.xpu.memory_allocated',
5386
- 'torch.xpu.memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory_reserved.html#torch.xpu.memory_reserved',
5387
- 'torch.xpu.memory_stats': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory_stats.html#torch.xpu.memory_stats',
5388
- 'torch.xpu.memory_stats_as_nested_dict': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory_stats_as_nested_dict.html#torch.xpu.memory_stats_as_nested_dict',
5389
5449
  'torch.xpu.random': 'https://pytorch.org/docs/stable/xpu.html#module-torch.xpu.random',
5390
- 'torch.xpu.reset_accumulated_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.xpu.reset_accumulated_memory_stats.html#torch.xpu.reset_accumulated_memory_stats',
5391
- 'torch.xpu.reset_peak_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.xpu.reset_peak_memory_stats.html#torch.xpu.reset_peak_memory_stats',
5392
5450
  'torch.xpu.seed': 'https://pytorch.org/docs/stable/generated/torch.xpu.seed.html#torch.xpu.seed',
5393
5451
  'torch.xpu.seed_all': 'https://pytorch.org/docs/stable/generated/torch.xpu.seed_all.html#torch.xpu.seed_all',
5394
5452
  'torch.xpu.set_device': 'https://pytorch.org/docs/stable/generated/torch.xpu.set_device.html#torch.xpu.set_device',
@@ -5398,5 +5456,15 @@ d = { 'settings': {'lib_path': 'nbdev_pytorch'},
5398
5456
  'torch.xpu.stream': 'https://pytorch.org/docs/stable/generated/torch.xpu.stream.html#torch.xpu.stream',
5399
5457
  'torch.xpu.streams': 'https://pytorch.org/docs/stable/xpu.html#module-torch.xpu.streams',
5400
5458
  'torch.xpu.synchronize': 'https://pytorch.org/docs/stable/generated/torch.xpu.synchronize.html#torch.xpu.synchronize'},
5459
+ 'torch.xpu.memory': { 'torch.xpu.memory.empty_cache': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.empty_cache.html#torch.xpu.memory.empty_cache',
5460
+ 'torch.xpu.memory.max_memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.max_memory_allocated.html#torch.xpu.memory.max_memory_allocated',
5461
+ 'torch.xpu.memory.max_memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.max_memory_reserved.html#torch.xpu.memory.max_memory_reserved',
5462
+ 'torch.xpu.memory.mem_get_info': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.mem_get_info.html#torch.xpu.memory.mem_get_info',
5463
+ 'torch.xpu.memory.memory_allocated': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.memory_allocated.html#torch.xpu.memory.memory_allocated',
5464
+ 'torch.xpu.memory.memory_reserved': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.memory_reserved.html#torch.xpu.memory.memory_reserved',
5465
+ 'torch.xpu.memory.memory_stats': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.memory_stats.html#torch.xpu.memory.memory_stats',
5466
+ 'torch.xpu.memory.memory_stats_as_nested_dict': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.memory_stats_as_nested_dict.html#torch.xpu.memory.memory_stats_as_nested_dict',
5467
+ 'torch.xpu.memory.reset_accumulated_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.reset_accumulated_memory_stats.html#torch.xpu.memory.reset_accumulated_memory_stats',
5468
+ 'torch.xpu.memory.reset_peak_memory_stats': 'https://pytorch.org/docs/stable/generated/torch.xpu.memory.reset_peak_memory_stats.html#torch.xpu.memory.reset_peak_memory_stats'},
5401
5469
  'torch.xpu.streams': { 'torch.xpu.streams.Event': 'https://pytorch.org/docs/stable/generated/torch.xpu.Event.html#torch.xpu.Event',
5402
5470
  'torch.xpu.streams.Stream': 'https://pytorch.org/docs/stable/generated/torch.xpu.Stream.html#torch.xpu.Stream'}}}
{nbdev_pytorch-0.0.2016.dist-info → nbdev_pytorch-0.0.2018.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nbdev-pytorch
3
- Version: 0.0.2016
3
+ Version: 0.0.2018
4
4
  Summary: nbdev docs lookup for PyTorch
5
5
  Home-page: https://github.com/fastai/nbdev-index/tree/master/
6
6
  Author: Jeremy Howard
nbdev_pytorch-0.0.2018.dist-info/RECORD ADDED
@@ -0,0 +1,9 @@
1
+ nbdev_pytorch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
+ nbdev_pytorch/_modidx.py,sha256=g-1XclneEFx7Ek-ee13VbZsz8gZ97_ZAR3irv2shGsk,958799
3
+ nbdev_pytorch/_nbdev.py,sha256=pszp1XJPRn4aqhJXSv8qChB-gCXbFSm8_jPIpak5IP8,308
4
+ nbdev_pytorch-0.0.2018.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
5
+ nbdev_pytorch-0.0.2018.dist-info/METADATA,sha256=b--8HGLA-c9K-XudFspUpcj0rJgfS3VwL54KHQ60P8Q,1176
6
+ nbdev_pytorch-0.0.2018.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
7
+ nbdev_pytorch-0.0.2018.dist-info/entry_points.txt,sha256=Mopx5xWWY0Rie1DuubmSXC-czc8-QmPDb3THWIlHoCA,48
8
+ nbdev_pytorch-0.0.2018.dist-info/top_level.txt,sha256=n0joXyL2rJVaevNQCfz-EPrISbaH52gShAiqXRagFRY,14
9
+ nbdev_pytorch-0.0.2018.dist-info/RECORD,,
nbdev_pytorch-0.0.2016.dist-info/RECORD DELETED
@@ -1,9 +0,0 @@
1
- nbdev_pytorch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- nbdev_pytorch/_modidx.py,sha256=LYGx67ozq6IElY4nTMhoqfVi9gW1JdcNL81kZPq922U,943436
3
- nbdev_pytorch/_nbdev.py,sha256=pszp1XJPRn4aqhJXSv8qChB-gCXbFSm8_jPIpak5IP8,308
4
- nbdev_pytorch-0.0.2016.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
5
- nbdev_pytorch-0.0.2016.dist-info/METADATA,sha256=yAvlZ_NIsTLQaDn2hnLzdd5EtT3m5i7VO-ihltpBRFQ,1176
6
- nbdev_pytorch-0.0.2016.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
7
- nbdev_pytorch-0.0.2016.dist-info/entry_points.txt,sha256=Mopx5xWWY0Rie1DuubmSXC-czc8-QmPDb3THWIlHoCA,48
8
- nbdev_pytorch-0.0.2016.dist-info/top_level.txt,sha256=n0joXyL2rJVaevNQCfz-EPrISbaH52gShAiqXRagFRY,14
9
- nbdev_pytorch-0.0.2016.dist-info/RECORD,,