Website Blog TensorPlay documentation

tensorplay API tensorplay._composite_funcs API tensorplay._einsum API tensorplay._shape_funcs API tensorplay._stax.api API tensorplay.accelerator API tensorplay.amp.autocast_mode API tensorplay.ao.nn.intrinsic API tensorplay.ao.nn.qat API tensorplay.ao.nn.quantized API tensorplay.ao.nn.quantized.dynamic API tensorplay.ao.pruning API tensorplay.ao.quantization API tensorplay.autograd API tensorplay.autograd.anomaly_mode API tensorplay.autograd.forward_ad API tensorplay.autograd.function API tensorplay.autograd.functional API tensorplay.autograd.grad_mode API tensorplay.autograd.gradcheck API tensorplay.autograd.graph API tensorplay.autograd.profiler_util API tensorplay.autograd.variable API tensorplay.backends.cpu API tensorplay.backends.cuda API tensorplay.backends.mkl API tensorplay.backends.nnpack API tensorplay.backends.openmp API tensorplay.compiler API tensorplay.cuda API tensorplay.cuda.gds API tensorplay.cuda.graph_annotations API tensorplay.cuda.graphs API tensorplay.cuda.green_contexts API tensorplay.cuda.memory API tensorplay.cuda.nccl API tensorplay.cuda.nvtx API tensorplay.cuda.profiler API tensorplay.cuda.random API tensorplay.cuda.streams API tensorplay.cuda.tunable API tensorplay.distributed.algorithms API tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook API tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks API tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook API tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook API tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks API tensorplay.distributed.algorithms.model_averaging API tensorplay.distributed.autograd API tensorplay.distributed.checkpoint API tensorplay.distributed.device_mesh API tensorplay.distributed.elastic.agent.server API tensorplay.distributed.elastic.events API tensorplay.distributed.elastic.metrics API tensorplay.distributed.elastic.multiprocessing API tensorplay.distributed.elastic.rendezvous API tensorplay.distributed.elastic.timer API tensorplay.distributed.elastic.utils API tensorplay.distributed.elastic.utils.data API tensorplay.distributed.fsdp API tensorplay.distributed.fsdp.sharded_grad_scaler API tensorplay.distributed.launcher API tensorplay.distributed.nn API tensorplay.distributed.nn.api API tensorplay.distributed.nn.functional API tensorplay.distributed.optim API tensorplay.distributed.pipelining API tensorplay.distributed.rpc API tensorplay.distributed.tensor API tensorplay.distributed.tensor.parallel API tensorplay.distributions API tensorplay.export API tensorplay.fft._frequency API tensorplay.fft._transforms API tensorplay.func API tensorplay.functional API tensorplay.futures API tensorplay.library API tensorplay.linalg API tensorplay.masked API tensorplay.monitor API tensorplay.multiprocessing API tensorplay.nested API tensorplay.nn.attention API tensorplay.nn.attention.bias API tensorplay.nn.attention.varlen API tensorplay.nn.functional API tensorplay.nn.init API tensorplay.nn.modules.flatten API tensorplay.nn.modules.lazy API tensorplay.nn.modules.module API tensorplay.nn.parameter API tensorplay.nn.utils.rnn API tensorplay.nn.utils.stateless API tensorplay.onnx API tensorplay.overrides API tensorplay.package API tensorplay.profiler API tensorplay.profiler.profiler API tensorplay.random API tensorplay.serialization API tensorplay.serialization.api API tensorplay.signal.windows API tensorplay.special._elementwise API tensorplay.special._error API tensorplay.special._gamma API tensorplay.stax API tensorplay.testing API tensorplay.utils.checkpoint API tensorplay.utils.data API tensorplay.utils.viz API

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Complex numbers Computation-graph visualization - tensorplay.utils.viz DDP Communication Hooks Distributed communication package - tensorplay.distributed DLPack Environment variables foreach operations Index Meta device Multiprocessing package - tensorplay.multiprocessing Patching batch norm Pruning - tensorplay.ao.pruning Python Module Index Quantization - tensorplay.ao.quantization Quantum simulation Reproducibility Search Size Spiking neural networks Storage Tensor attributes Tensor views tensorplay tensorplay._stax tensorplay.accelerator tensorplay.audio tensorplay.backends tensorplay.compiler tensorplay.cuda tensorplay.cuda.tunable tensorplay.distributed.algorithms tensorplay.distributed.autograd tensorplay.distributed.checkpoint tensorplay.distributed.device_mesh tensorplay.distributed.elastic tensorplay.distributed.fsdp tensorplay.distributed.fsdp.fully_shard tensorplay.distributed.nn tensorplay.distributed.optim tensorplay.distributed.pipelining tensorplay.distributed.rpc tensorplay.distributed.tensor tensorplay.distributed.tensor.parallel tensorplay.distributions tensorplay.export tensorplay.fft tensorplay.func tensorplay.func API reference tensorplay.func whirlwind tour tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.masked tensorplay.monitor tensorplay.nested tensorplay.nn tensorplay.nn.attention tensorplay.nn.attention.bias tensorplay.nn.attention.omni_attention tensorplay.nn.attention.varlen tensorplay.nn.functional tensorplay.nn.init tensorplay.onnx tensorplay.optim tensorplay.overrides tensorplay.package tensorplay.profiler tensorplay.random tensorplay.serialization tensorplay.signal tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.testing tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Type information UX limitations

API reference

Documentation/API reference

latest (dev)

latest (dev)2026-10-15

Copy

Latest development documentation · Updated 2026-10-15

tensorplay.serialization API

Functions 21

add_safe_globals

functionFull reference ↗

tensorplay.serialization.add_safe_globals(values) → None[source]

clear_safe_globals

functionFull reference ↗

tensorplay.serialization.clear_safe_globals() → None[source]

convert_model

functionFull reference ↗

tensorplay.serialization.convert_model(model_dir, output_dir=None, **kwargs): Convert a model directory or a supported checkpoint into MEGA.

convert_to_mega

functionFull reference ↗

tensorplay.serialization.convert_to_mega(model_dir, output_dir=None, **kwargs)[source]: Convert a model directory or a supported checkpoint into MEGA.

default_restore_location

functionFull reference ↗

tensorplay.serialization.default_restore_location(storage, location)[source]

get_crc32_options

functionFull reference ↗

tensorplay.serialization.get_crc32_options() → bool[source]

get_default_load_endianness

functionFull reference ↗

tensorplay.serialization.get_default_load_endianness() → LoadEndianness | None[source]

get_default_mmap_options

functionFull reference ↗

tensorplay.serialization.get_default_mmap_options() → int | None[source]

get_safe_globals

functionFull reference ↗

tensorplay.serialization.get_safe_globals() → list[Any][source]

get_unsafe_globals_in_checkpoint

functionFull reference ↗

tensorplay.serialization.get_unsafe_globals_in_checkpoint(f) → list[str][source]

inspect_checkpoint

functionFull reference ↗

tensorplay.serialization.inspect_checkpoint(f, *, verify_checksums: bool = False) → dict[source]

load

functionFull reference ↗

tensorplay.serialization.load(f, map_location=None, pickle_module=None, *, weights_only: bool = True, mmap: bool | None = None, **pickle_load_args) → Any[source]

location_tag

functionFull reference ↗

tensorplay.serialization.location_tag(storage)[source]

normalize_storage_type

functionFull reference ↗

tensorplay.serialization.normalize_storage_type(storage_type)[source]

parse_mega_header

functionFull reference ↗

tensorplay.serialization.parse_mega_header(path: str | PathLike[str]) → dict[source]: Read MEGA metadata and tensor ranges without materializing tensors.

register_package

functionFull reference ↗

tensorplay.serialization.register_package(priority: int, tagger, deserializer)[source]

resolve_map_location

functionFull reference ↗

tensorplay.serialization.resolve_map_location(map_location: Any, location: str, nbytes: int = 0)[source]

Resolve one saved storage location.

None means that the saved location should be used by the caller. A callable may return a device, a destination tensor, or None.

save

functionFull reference ↗

tensorplay.serialization.save(obj: Any, f, pickle_module=<module 'pickle' from '/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/pickle.py'>, pickle_protocol: int = 2, _use_new_zipfile_serialization: bool = True, _disable_byteorder_record: bool = False, *, metadata: Mapping[str, ~typing.Any] | None=None, alignment: int = 4096, checksum: str = 'crc32') → None[source]

set_crc32_options

functionFull reference ↗

tensorplay.serialization.set_crc32_options(compute_crc32: bool)[source]

set_default_load_endianness

functionFull reference ↗

tensorplay.serialization.set_default_load_endianness(endianness)[source]

storage_to_tensor_type

functionFull reference ↗

tensorplay.serialization.storage_to_tensor_type(storage)[source]

Classes 4

LoadEndianness

classFull reference ↗

class tensorplay.serialization.LoadEndianness(*values)[source]

safe_globals

classFull reference ↗

class tensorplay.serialization.safe_globals(values)[source]

set_default_mmap_options

classFull reference ↗

class tensorplay.serialization.set_default_mmap_options(flags: int)[source]

skip_data

classFull reference ↗

class tensorplay.serialization.skip_data(materialize_fake_tensors: bool = False)[source]

Attributes 1

StorageType

attributeFull reference ↗

tensorplay.serialization.StorageType: alias of _StorageType

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.random API

Complete API reference for tensorplay.random, including signatures, parameters, examples and members.

tensorplay.serialization.api API

Complete API reference for tensorplay.serialization.api, including signatures, parameters, examples and members.

On this page

Functions 21 add_safe_globals clear_safe_globals convert_model convert_to_mega default_restore_location get_crc32_options get_default_load_endianness get_default_mmap_options get_safe_globals get_unsafe_globals_in_checkpoint inspect_checkpoint load location_tag normalize_storage_type parse_mega_header register_package resolve_map_location save set_crc32_options set_default_load_endianness storage_to_tensor_type Classes 4 LoadEndianness safe_globals set_default_mmap_options skip_data Attributes 1 StorageType