mlx-cluster 0.0.1__cp311-cp311-macosx_14_0_arm64.whl

mlx_cluster/__init__.py

@@ -0,0 +1,4 @@
+import mlx.core as mx
+
+from ._ext import random_walk
+from ._ext import rejection_sampling

mlx_cluster-0.0.1.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 vinayhpandya
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

mlx_cluster-0.0.1.dist-info/METADATA

@@ -0,0 +1,74 @@
+Metadata-Version: 2.1
+Name: mlx_cluster
+Version: 0.0.1
+Summary: C++ and Metal extensions for MLX CTC Loss
+Author-email: Vinay Pandya <vinayharshadpandya27@gmail.com>
+Project-URL: Homepage, https://github.com/vinayhpandya/mlx_cluster
+Project-URL: Issues, https://github.com/vinayhpandya/mlx_cluster/Issues
+Classifier: Development Status :: 3 - Alpha
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: C++
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Provides-Extra: dev
+Provides-Extra: test
+Requires-Dist: torch-geometric ; extra == 'test'
+Requires-Dist: pytest ; extra == 'test'
+
+# mlx_cluster
+
+A C++ extension for generating ramdom walks for Homogeneous graphs using mlx
+
+## Installation
+
+To install the necessary dependencies:
+
+Clone the repositories:
+```bash
+git clone https://github.com/vinayhpandya/mlx_cluster.git
+```
+
+After cloning the repository install library using 
+
+```bash
+python setup.py build_ext -j8 --inplace
+```
+
+for testing purposes you need to have `mlx-graphs` installed
+
+## Usage
+
+
+```
+from mlx_graphs.utils.sorting import sort_edge_index
+from mlx_graphs.loaders import Dataloader
+from mlx_graphs_extension import random_walk
+
+
+cora_dataset = PlanetoidDataset(name="cora", base_dir="~")
+start = mx.arange(0, 1000)
+start_time = time.time()
+edge_index = cora_dataset.graphs[0].edge_index
+num_nodes = cora_dataset.graphs[0].num_nodes
+sorted_edge_index = sort_edge_index(edge_index=edge_index)
+row_mlx = sorted_edge_index[0][0]
+col_mlx = sorted_edge_index[0][1]
+unique_vals, counts_mlx = np.unique(np.array(row_mlx, copy=False), return_counts=True)
+cum_sum_mlx = counts_mlx.cumsum()
+rand = mx.random.uniform(shape=[start.shape[0], 100])
+row_ptr_mlx = mx.concatenate([mx.array([0]), mx.array(cum_sum_mlx)])
+random_walk(row_ptr_mlx, col_mlx, start, rand, 1000, stream = mx.gpu)
+```
+
+## TODO
+
+- [x] Add metal shaders to optimize the code
+- [ ] Benchmark random walk against different frameworks
+- [ ] Add more algorithms
+
+## Credits:
+
+torch_cluster random walk implementation : [random_walk](https://github.com/rusty1s/pytorch_cluster/blob/master/csrc/cpu/rw_cpu.cpp)

mlx_cluster-0.0.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+mlx_cluster/__init__.py,sha256=PNgjmiw5F4DRhwCXwtJH5B22C-eB6dVD8Nqctj_3kaY,90
+mlx_cluster/_ext.cpython-311-darwin.so,sha256=_puyYR4x5DK3RtvrOwvkbpwOKGm5d2WqqIfCTGQIPNA,132360
+mlx_cluster/libmlx_cluster.dylib,sha256=b7YcEY6lj7bdco_9P-EOU5uccRpa6j17MrX7Z9ljtTU,68808
+mlx_cluster/mlx_cluster.metallib,sha256=ysmhLDXKt7OvwRS4cQZhn9c1Mwxji29gpCvFKwh3Df4,5416
+mlx_cluster-0.0.1.dist-info/LICENSE,sha256=7ixsoVuroKzGl84-ZV90CDH6bQrQF2UXvgnTqtW4Zb4,1069
+mlx_cluster-0.0.1.dist-info/METADATA,sha256=bGWxSL6z-m4faVhSxNx2Z5G78aoBemy7BGXJqMfQfyA,2254
+mlx_cluster-0.0.1.dist-info/WHEEL,sha256=j3-yksFZs2SDtXJbm_EOXsPslt0JPXMfrvzk2bCbj2A,109
+mlx_cluster-0.0.1.dist-info/top_level.txt,sha256=sERi0kZuQnKmsHnqOkAda1kkBXdQAwV25Y9L-ATj110,12
+mlx_cluster-0.0.1.dist-info/RECORD,,

mlx_cluster-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.1.0)
+Root-Is-Purelib: false
+Tag: cp311-cp311-macosx_14_0_arm64
+

mlx_cluster-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+mlx_cluster