Index A | B | C | D | E | F | G | I | K | L | M | N | O | P | Q | R | S | T | U | V | W A active_count (mlx_lattice.core.coords.alignment.SparseAlignment property) (mlx_lattice.core.coords.occupancy.OccupancyExpansion property) (mlx_lattice.core.coords.occupancy.SparseOccupancy property) (mlx_lattice.core.coords.quantization.SparseQuantization property) (mlx_lattice.core.coords.set_ops.CoordinateSet property) (mlx_lattice.core.tensor.SparseTensor property) active_rows (mlx_lattice.core.coords.alignment.SparseAlignment attribute) (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) (mlx_lattice.core.coords.occupancy.SparseOccupancy attribute) (mlx_lattice.core.coords.quantization.SparseQuantization attribute) (mlx_lattice.core.coords.set_ops.CoordinateSet attribute) (mlx_lattice.core.tensor.SparseTensor attribute) active_rows() (mlx_lattice.core.coords.manager.CoordinateManager method) align_sparse() (in module mlx_lattice.ops.tensor) astype() (mlx_lattice.core.tensor.SparseTensor method) avg_pool3d() (in module mlx_lattice.ops.pool) AvgPool3d (class in mlx_lattice.nn.pool) B backend_info() (in module mlx_lattice) batch_counts (mlx_lattice.core.tensor.SparseTensor attribute) batch_indices (mlx_lattice.core.tensor.SparseTensor property) batch_norm() (in module mlx_lattice.ops.feature) batch_rows (mlx_lattice.core.tensor.SparseTensor property) BatchNorm (class in mlx_lattice.nn.feature) biases (mlx_lattice.core.quantized.QuantizedWeight attribute) bits (mlx_lattice.core.quantized.QuantizedWeight attribute) build_generative_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_knn_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_point_voxel_map() (in module mlx_lattice.core.coords.quantization) (in module mlx_lattice.ops.quantization) build_radius_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_relation_implicit_gemm_view() (in module mlx_lattice.core.coords.builders) build_sparse_alignment() (in module mlx_lattice.core.coords.alignment) build_submanifold_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_target_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_transposed_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) C capacity (mlx_lattice.core.coords.alignment.SparseAlignment property) (mlx_lattice.core.coords.occupancy.OccupancyExpansion property) (mlx_lattice.core.coords.occupancy.SparseOccupancy property) (mlx_lattice.core.coords.quantization.SparseQuantization property) (mlx_lattice.core.coords.set_ops.CoordinateSet property) (mlx_lattice.core.relations.views.NeighborEdges property) (mlx_lattice.core.relations.views.RelationEdges property) (mlx_lattice.core.tensor.SparseTensor property) cat() (in module mlx_lattice.ops.tensor) channels (mlx_lattice.core.tensor.SparseTensor property) child_coords_from_indices() (in module mlx_lattice.core.coords.occupancy) child_indices (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) contains_coords() (in module mlx_lattice.core.coords.set_ops) contract (mlx_lattice.core.relations.views.KernelRelation attribute) Conv3d (class in mlx_lattice.nn.conv) conv3d() (in module mlx_lattice.ops.conv) conv_transpose3d() (in module mlx_lattice.ops.conv) ConvTranspose3d (class in mlx_lattice.nn.conv) coord_key (mlx_lattice.core.tensor.SparseTensor attribute) coord_manager (mlx_lattice.core.tensor.SparseTensor attribute) CoordinateManager (class in mlx_lattice.core.coords.manager) CoordinateMapKey (class in mlx_lattice.core.coords.manager) CoordinateOrdering (class in mlx_lattice.core.coords.ordering) CoordinateSet (class in mlx_lattice.core.coords.set_ops) coords (mlx_lattice.core.coords.alignment.SparseAlignment attribute) (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) (mlx_lattice.core.coords.occupancy.SparseOccupancy attribute) (mlx_lattice.core.coords.ordering.CoordinateOrdering attribute) (mlx_lattice.core.coords.quantization.SparseQuantization attribute) (mlx_lattice.core.coords.set_ops.CoordinateSet attribute) (mlx_lattice.core.tensor.SparseTensor attribute) coords() (mlx_lattice.core.coords.manager.CoordinateManager method) counts (mlx_lattice.core.coords.quantization.SparseQuantization attribute) (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) crop() (in module mlx_lattice.ops.tensor) D decomposed_coordinates (mlx_lattice.core.tensor.SparseTensor property) decomposed_features (mlx_lattice.core.tensor.SparseTensor property) dequantize_weight() (in module mlx_lattice.core.quantized) devoxelize() (in module mlx_lattice.ops.quantization) dilation (mlx_lattice.core.relations.specs.KernelSpec attribute) distances (mlx_lattice.core.relations.views.NeighborRelation attribute) downsample_coords() (in module mlx_lattice.core.coords.set_ops) Dropout (class in mlx_lattice.nn.feature) dropout() (in module mlx_lattice.ops.feature) dtype (mlx_lattice.core.tensor.SparseTensor property) E edge_capacity (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation property) edge_count (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation property) (mlx_lattice.core.relations.views.SparseRelationContract property) edge_ids (mlx_lattice.core.relations.views.RelationCSRView attribute) edges (mlx_lattice.core.relations.views.KernelRelation attribute) (mlx_lattice.core.relations.views.NeighborRelation attribute) F feats (mlx_lattice.core.tensor.SparseTensor attribute) from_conv() (mlx_lattice.nn.quantized_conv.QuantizedConv3d class method) (mlx_lattice.nn.quantized_conv.QuantizedConvTranspose3d class method) (mlx_lattice.nn.quantized_conv.QuantizedGenerativeConvTranspose3d class method) (mlx_lattice.nn.quantized_conv.QuantizedSubmConv3d class method) from_linear() (mlx_lattice.nn.quantized_feature.QuantizedLinear class method) G gather_aligned_features() (in module mlx_lattice.ops.tensor) gather_neighbor_features() (in module mlx_lattice.ops.relations) GELU (class in mlx_lattice.nn.feature) gelu() (in module mlx_lattice.ops.feature) generative_conv_transpose3d() (in module mlx_lattice.ops.conv) generative_kernel_relation() (in module mlx_lattice.ops.relations) generative_relation() (mlx_lattice.core.coords.manager.CoordinateManager method) GenerativeConvTranspose3d (class in mlx_lattice.nn.conv) global_avg_pool() (in module mlx_lattice.ops.pool) global_max_pool() (in module mlx_lattice.ops.pool) global_sum_pool() (in module mlx_lattice.ops.pool) GlobalAvgPool (class in mlx_lattice.nn.pool) GlobalMaxPool (class in mlx_lattice.nn.pool) GlobalSumPool (class in mlx_lattice.nn.pool) group_size (mlx_lattice.core.quantized.QuantizedWeight attribute) I id (mlx_lattice.core.coords.manager.CoordinateMapKey attribute) implicit_gemm (mlx_lattice.core.relations.views.KernelRelation attribute) in_channels (mlx_lattice.core.quantized.QuantizedWeight attribute) in_edge_ids (mlx_lattice.core.relations.views.KernelRelation property) in_row_offsets (mlx_lattice.core.relations.views.KernelRelation property) in_rows (mlx_lattice.core.relations.views.RelationEdges attribute) in_view (mlx_lattice.core.relations.views.KernelRelation property) input_csr (mlx_lattice.core.relations.views.KernelRelation attribute) insert_coords() (mlx_lattice.core.coords.manager.CoordinateManager method) interpolate_point_features() (in module mlx_lattice.core.coords.quantization) (in module mlx_lattice.ops.quantization) intersection_coords() (in module mlx_lattice.core.coords.set_ops) inverse_map() (in module mlx_lattice.core.coords.set_ops) (mlx_lattice.core.coords.manager.CoordinateManager method) inverse_rows (mlx_lattice.core.coords.ordering.CoordinateOrdering attribute) (mlx_lattice.core.coords.quantization.SparseQuantization attribute) is_centered_submanifold (mlx_lattice.core.relations.specs.KernelSpec property) is_pointwise (mlx_lattice.core.quantized.QuantizedWeight property) (mlx_lattice.core.relations.specs.KernelSpec property) K kernel_csr (mlx_lattice.core.relations.views.KernelRelation attribute) kernel_edge_ids (mlx_lattice.core.relations.views.KernelRelation property) kernel_ids (mlx_lattice.core.relations.views.RelationEdges attribute) kernel_offsets (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) kernel_offsets() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) kernel_row_offsets (mlx_lattice.core.relations.views.KernelRelation property) kernel_size (mlx_lattice.core.quantized.QuantizedWeight attribute) kernel_view (mlx_lattice.core.relations.views.KernelRelation property) KernelRelation (class in mlx_lattice.core.relations.views) KernelSpec (class in mlx_lattice.core.relations.specs) kind (mlx_lattice.core.relations.views.SparseRelationContract attribute) knn_relation() (in module mlx_lattice.ops.relations) L layer_norm() (in module mlx_lattice.ops.feature) LayerNorm (class in mlx_lattice.nn.feature) layout (mlx_lattice.core.quantized.QuantizedWeight attribute) leaky_relu() (in module mlx_lattice.ops.feature) LeakyReLU (class in mlx_lattice.nn.feature) lhs_rows (mlx_lattice.core.coords.alignment.SparseAlignment attribute) Linear (class in mlx_lattice.nn.feature) linear() (in module mlx_lattice.ops.feature) lookup_coords() (in module mlx_lattice.core.coords.set_ops) M manager_id (mlx_lattice.core.coords.manager.CoordinateMapKey attribute) max_neighbors (mlx_lattice.core.relations.views.NeighborRelation attribute) max_pool3d() (in module mlx_lattice.ops.pool) MaxPool3d (class in mlx_lattice.nn.pool) mlx_lattice.core.coords.alignment module mlx_lattice.core.coords.builders module mlx_lattice.core.coords.manager module mlx_lattice.core.coords.occupancy module mlx_lattice.core.coords.ordering module mlx_lattice.core.coords.quantization module mlx_lattice.core.coords.set_ops module mlx_lattice.core.quantized module mlx_lattice.core.relations.specs module mlx_lattice.core.relations.views module mlx_lattice.core.tensor module mlx_lattice.core.types module mlx_lattice.nn.conv module mlx_lattice.nn.feature module mlx_lattice.nn.pool module mlx_lattice.nn.quantized_conv module mlx_lattice.nn.quantized_feature module mlx_lattice.ops._quantized module mlx_lattice.ops._relation_exec module mlx_lattice.ops.conv module mlx_lattice.ops.entropy module mlx_lattice.ops.feature module mlx_lattice.ops.pool module mlx_lattice.ops.quantization module mlx_lattice.ops.relations module mlx_lattice.ops.tensor module module mlx_lattice.core.coords.alignment mlx_lattice.core.coords.builders mlx_lattice.core.coords.manager mlx_lattice.core.coords.occupancy mlx_lattice.core.coords.ordering mlx_lattice.core.coords.quantization mlx_lattice.core.coords.set_ops mlx_lattice.core.quantized mlx_lattice.core.relations.specs mlx_lattice.core.relations.views mlx_lattice.core.tensor mlx_lattice.core.types mlx_lattice.nn.conv mlx_lattice.nn.feature mlx_lattice.nn.pool mlx_lattice.nn.quantized_conv mlx_lattice.nn.quantized_feature mlx_lattice.ops._quantized mlx_lattice.ops._relation_exec mlx_lattice.ops.conv mlx_lattice.ops.entropy mlx_lattice.ops.feature mlx_lattice.ops.pool mlx_lattice.ops.quantization mlx_lattice.ops.relations mlx_lattice.ops.tensor morton_codes() (in module mlx_lattice.core.coords.ordering) morton_order() (in module mlx_lattice.core.coords.ordering) morton_sort_coords() (in module mlx_lattice.core.coords.ordering) N n_in_capacity (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) n_kernels (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) n_out_capacity (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) n_query_capacity (mlx_lattice.core.relations.views.NeighborRelation attribute) n_source_capacity (mlx_lattice.core.relations.views.NeighborRelation attribute) NativeImplicitGemmView (in module mlx_lattice.core.coords.builders) NativeKernelRelation (in module mlx_lattice.core.coords.builders) NativeNeighborRelation (in module mlx_lattice.core.coords.builders) nbytes (mlx_lattice.core.quantized.QuantizedWeight property) neighbor_ids (mlx_lattice.core.relations.views.NeighborEdges attribute) NeighborEdges (class in mlx_lattice.core.relations.views) NeighborRelation (class in mlx_lattice.core.relations.views) normalized_cdf() (in module mlx_lattice.ops.entropy) O occupancy (mlx_lattice.core.coords.occupancy.SparseOccupancy attribute) occupancy_downsample() (in module mlx_lattice.core.coords.occupancy) occupancy_expand() (in module mlx_lattice.core.coords.occupancy) OccupancyExpansion (class in mlx_lattice.core.coords.occupancy) order (mlx_lattice.core.coords.ordering.CoordinateOrdering attribute) out_channels (mlx_lattice.core.quantized.QuantizedWeight attribute) out_coords (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) out_count (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract property) out_in_map (mlx_lattice.core.relations.views.RelationImplicitGemmView attribute) out_rows (mlx_lattice.core.relations.views.RelationEdges attribute) out_view (mlx_lattice.core.relations.views.KernelRelation property) output_csr (mlx_lattice.core.relations.views.KernelRelation attribute) owns() (mlx_lattice.core.coords.manager.CoordinateManager method) P padding (mlx_lattice.core.relations.specs.KernelSpec attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) parent_rows (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) point_count (mlx_lattice.core.coords.quantization.PointVoxelMap property) PointVoxelInterpolation (in module mlx_lattice.core.coords.quantization) PointVoxelMap (class in mlx_lattice.core.coords.quantization) Pool3d (class in mlx_lattice.nn.pool) pool3d() (in module mlx_lattice.ops.pool) prune() (in module mlx_lattice.ops.tensor) prune_mask() (in module mlx_lattice.ops.tensor) Q quantize_weight() (in module mlx_lattice.core.quantized) quantized_matmul() (in module mlx_lattice.ops._quantized) QuantizedConv3d (class in mlx_lattice.nn.quantized_conv) QuantizedConvTranspose3d (class in mlx_lattice.nn.quantized_conv) QuantizedGenerativeConvTranspose3d (class in mlx_lattice.nn.quantized_conv) QuantizedLinear (class in mlx_lattice.nn.quantized_feature) QuantizedSubmConv3d (class in mlx_lattice.nn.quantized_conv) QuantizedWeight (class in mlx_lattice.core.quantized) query_count (mlx_lattice.core.relations.views.NeighborRelation property) query_rows (mlx_lattice.core.relations.views.NeighborEdges attribute) R radius_relation() (in module mlx_lattice.ops.relations) range_decode() (in module mlx_lattice.ops.entropy) range_decode_from_prob() (in module mlx_lattice.ops.entropy) range_encode() (in module mlx_lattice.ops.entropy) range_encode_from_prob() (in module mlx_lattice.ops.entropy) rans_decode_from_prob() (in module mlx_lattice.ops.entropy) rans_encode_from_prob() (in module mlx_lattice.ops.entropy) RelationCSRView (class in mlx_lattice.core.relations.views) RelationEdges (class in mlx_lattice.core.relations.views) RelationImplicitGemmView (class in mlx_lattice.core.relations.views) RelationSortedImplicitGemmView (class in mlx_lattice.core.relations.views) RelationView (in module mlx_lattice.core.relations.views) ReLU (class in mlx_lattice.nn.feature) relu() (in module mlx_lattice.ops.feature) reorder_rows (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) replace() (mlx_lattice.core.tensor.SparseTensor method) replace_feature() (in module mlx_lattice.ops.tensor) require_implicit_gemm() (mlx_lattice.core.relations.views.KernelRelation method) require_sorted_implicit_gemm() (mlx_lattice.core.relations.views.KernelRelation method) reuse_coords_from() (mlx_lattice.core.tensor.SparseTensor method) rhs_rows (mlx_lattice.core.coords.alignment.SparseAlignment attribute) rms_norm() (in module mlx_lattice.ops.feature) RMSNorm (class in mlx_lattice.nn.feature) row_masks (mlx_lattice.core.relations.views.RelationImplicitGemmView attribute) row_offsets (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation attribute) (mlx_lattice.core.relations.views.RelationCSRView attribute) rows (mlx_lattice.core.coords.quantization.PointVoxelMap attribute) S same_coords() (mlx_lattice.core.tensor.SparseTensor method) scales (mlx_lattice.core.quantized.QuantizedWeight attribute) shape (mlx_lattice.core.tensor.SparseTensor property) Sigmoid (class in mlx_lattice.nn.feature) sigmoid() (in module mlx_lattice.ops.feature) SiLU (class in mlx_lattice.nn.feature) silu() (in module mlx_lattice.ops.feature) size (mlx_lattice.core.relations.specs.KernelSpec attribute) Softplus (class in mlx_lattice.nn.feature) softplus() (in module mlx_lattice.ops.feature) sorted_implicit_gemm (mlx_lattice.core.relations.views.KernelRelation attribute) sorted_kv_out_in_map (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) sorted_out_in_map (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) source_active_rows (mlx_lattice.core.relations.views.SparseRelationContract attribute) source_coords (mlx_lattice.core.relations.views.SparseRelationContract attribute) source_rows (mlx_lattice.core.relations.views.NeighborEdges attribute) sparse_add() (in module mlx_lattice.ops.tensor) sparse_binary_op() (in module mlx_lattice.ops.tensor) sparse_cat_aligned() (in module mlx_lattice.ops.tensor) sparse_collate() (in module mlx_lattice.ops.tensor) sparse_conv_features_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_conv_features_sorted_direct_reference_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_conv_features_sorted_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_maximum() (in module mlx_lattice.ops.tensor) sparse_minimum() (in module mlx_lattice.ops.tensor) sparse_mul() (in module mlx_lattice.ops.tensor) sparse_pool_features_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_quantize() (in module mlx_lattice.core.coords.quantization) sparse_quantized_conv_features_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_sub() (in module mlx_lattice.ops.tensor) SparseAlignment (class in mlx_lattice.core.coords.alignment) SparseJoin (in module mlx_lattice.core.coords.alignment) (in module mlx_lattice.ops.tensor) SparseOccupancy (class in mlx_lattice.core.coords.occupancy) SparseQuantization (class in mlx_lattice.core.coords.quantization) SparseRelationContract (class in mlx_lattice.core.relations.views) SparseTensor (class in mlx_lattice.core.tensor) storage_in_channels (mlx_lattice.core.quantized.QuantizedWeight property) stride (mlx_lattice.core.coords.manager.CoordinateMapKey attribute) (mlx_lattice.core.relations.specs.KernelSpec attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) (mlx_lattice.core.tensor.SparseTensor attribute) subm_conv3d() (in module mlx_lattice.ops.conv) submanifold_kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) SubmConv3d (class in mlx_lattice.nn.conv) sum_pool3d() (in module mlx_lattice.ops.pool) SumPool3d (class in mlx_lattice.nn.pool) T Tanh (class in mlx_lattice.nn.feature) tanh() (in module mlx_lattice.ops.feature) target_active_rows (mlx_lattice.core.relations.views.SparseRelationContract attribute) target_coords (mlx_lattice.core.relations.views.SparseRelationContract attribute) target_kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) tile_masks (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) to_quantized() (mlx_lattice.nn.conv.Conv3d method) (mlx_lattice.nn.conv.ConvTranspose3d method) (mlx_lattice.nn.conv.GenerativeConvTranspose3d method) (mlx_lattice.nn.conv.SubmConv3d method) (mlx_lattice.nn.feature.Linear method) topk_rows() (in module mlx_lattice.ops.tensor) transposed_kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) triple() (in module mlx_lattice.core.types) U union_coords() (in module mlx_lattice.core.coords.set_ops) V volume (mlx_lattice.core.relations.specs.KernelSpec property) voxelize() (in module mlx_lattice.ops.quantization) voxelize_features() (in module mlx_lattice.core.coords.quantization) voxelize_with_quantization() (in module mlx_lattice.ops.quantization) VoxelReduction (in module mlx_lattice.core.coords.quantization) W weight (mlx_lattice.core.quantized.QuantizedWeight attribute) weights (mlx_lattice.core.coords.quantization.PointVoxelMap attribute)