SparseCompressedΒΆ Layout constant PyTorchJAX (Core) PyTorchAPI: torch.sparse_compressedStrategy: Direct MappingOfficial Docs βJAX (Core)API: βStrategy: Custom / Partial