fastkmeans 0.4.0__tar.gz → 0.5.0__tar.gz

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.
Files changed (13) hide show
  1. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/PKG-INFO +15 -2
  2. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/README.md +14 -1
  3. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans/__init__.py +1 -1
  4. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans/kmeans.py +2 -0
  5. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans/triton_kernels.py +2 -0
  6. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans.egg-info/PKG-INFO +15 -2
  7. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/pyproject.toml +1 -1
  8. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/LICENSE +0 -0
  9. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans.egg-info/SOURCES.txt +0 -0
  10. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans.egg-info/dependency_links.txt +0 -0
  11. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans.egg-info/requires.txt +0 -0
  12. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans.egg-info/top_level.txt +0 -0
  13. {fastkmeans-0.4.0 → fastkmeans-0.5.0}/setup.cfg +0 -0
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: fastkmeans
3
- Version: 0.4.0
3
+ Version: 0.5.0
4
4
  Summary: Add your description here
5
5
  Author-email: Ben Clavié <bc@answer.ai>, Benjamin Warner <bw@answer.ai>
6
6
  Maintainer-email: Ben Clavié <bc@answer.ai>, Benjamin Warner <bw@answer.ai>
@@ -14,8 +14,9 @@ Dynamic: license-file
14
14
 
15
15
  # fastkmeans
16
16
 
17
- ![Python Versions](https://img.shields.io/badge/Python-3.8_3.9_3.10_3.11_3.12_3.13-blue)
17
+ ![Python Versions](https://img.shields.io/badge/Python-3.9_3.10_3.11_3.12_3.13-blue)
18
18
  [![Twitter Follow](https://img.shields.io/twitter/follow/bclavie?style=social)](https://twitter.com/bclavie)
19
+ [![Twitter Follow](https://img.shields.io/twitter/follow/benjamin_warner?style=social)](https://twitter.com/benjamin_warner)
19
20
  <!-- [![Downloads](https://static.pepy.tech/badge/fastkmeans/month)](https://pepy.tech/project/fastkmeans) -->
20
21
 
21
22
  _A fast and efficient k-means implementation for PyTorch, with support for GPU and CPU._
@@ -96,3 +97,15 @@ Then, run the benchmark script:
96
97
  ```bash
97
98
  CUDA_VISIBLE_DEVICES=0 python speedbench.py --do-faiss --do-fastkmeans --do-fastkmeans-triton --do-evals
98
99
  ```
100
+
101
+ ### Citation
102
+
103
+ If you use fastmeans and want to/need to cite it in your work, please feel free to use the citation below:
104
+
105
+ ```bibtex
106
+ @misc{fastkmeans2025,
107
+ author = {Benjamin Clavié and Benjamin Warner},
108
+ title = {fastkmeans: Accelerated KMeans Clustering in PyTorch and Triton},
109
+ year = {2025},
110
+ howpublished = {\url{https://github.com/AnswerDotAI/fastkmeans/}}
111
+ }
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/README.md RENAMED
@@ -1,7 +1,8 @@
1
1
  # fastkmeans
2
2
 
3
- ![Python Versions](https://img.shields.io/badge/Python-3.8_3.9_3.10_3.11_3.12_3.13-blue)
3
+ ![Python Versions](https://img.shields.io/badge/Python-3.9_3.10_3.11_3.12_3.13-blue)
4
4
  [![Twitter Follow](https://img.shields.io/twitter/follow/bclavie?style=social)](https://twitter.com/bclavie)
5
+ [![Twitter Follow](https://img.shields.io/twitter/follow/benjamin_warner?style=social)](https://twitter.com/benjamin_warner)
5
6
  <!-- [![Downloads](https://static.pepy.tech/badge/fastkmeans/month)](https://pepy.tech/project/fastkmeans) -->
6
7
 
7
8
  _A fast and efficient k-means implementation for PyTorch, with support for GPU and CPU._
@@ -82,3 +83,15 @@ Then, run the benchmark script:
82
83
  ```bash
83
84
  CUDA_VISIBLE_DEVICES=0 python speedbench.py --do-faiss --do-fastkmeans --do-fastkmeans-triton --do-evals
84
85
  ```
86
+
87
+ ### Citation
88
+
89
+ If you use fastmeans and want to/need to cite it in your work, please feel free to use the citation below:
90
+
91
+ ```bibtex
92
+ @misc{fastkmeans2025,
93
+ author = {Benjamin Clavié and Benjamin Warner},
94
+ title = {fastkmeans: Accelerated KMeans Clustering in PyTorch and Triton},
95
+ year = {2025},
96
+ howpublished = {\url{https://github.com/AnswerDotAI/fastkmeans/}}
97
+ }
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans/__init__.py RENAMED
@@ -1,4 +1,4 @@
1
1
  from .kmeans import FastKMeans
2
2
 
3
3
  __all__ = ["FastKMeans"]
4
- __version__ = "0.4.0"
4
+ __version__ = "0.5.0"
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans/kmeans.py RENAMED
@@ -1,3 +1,5 @@
1
+ from __future__ import annotations
2
+
1
3
  import time
2
4
 
3
5
  import torch
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans/triton_kernels.py RENAMED
@@ -1,3 +1,5 @@
1
+ from __future__ import annotations
2
+
1
3
  from contextlib import contextmanager
2
4
 
3
5
  import torch
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/fastkmeans.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: fastkmeans
3
- Version: 0.4.0
3
+ Version: 0.5.0
4
4
  Summary: Add your description here
5
5
  Author-email: Ben Clavié <bc@answer.ai>, Benjamin Warner <bw@answer.ai>
6
6
  Maintainer-email: Ben Clavié <bc@answer.ai>, Benjamin Warner <bw@answer.ai>
@@ -14,8 +14,9 @@ Dynamic: license-file
14
14
 
15
15
  # fastkmeans
16
16
 
17
- ![Python Versions](https://img.shields.io/badge/Python-3.8_3.9_3.10_3.11_3.12_3.13-blue)
17
+ ![Python Versions](https://img.shields.io/badge/Python-3.9_3.10_3.11_3.12_3.13-blue)
18
18
  [![Twitter Follow](https://img.shields.io/twitter/follow/bclavie?style=social)](https://twitter.com/bclavie)
19
+ [![Twitter Follow](https://img.shields.io/twitter/follow/benjamin_warner?style=social)](https://twitter.com/benjamin_warner)
19
20
  <!-- [![Downloads](https://static.pepy.tech/badge/fastkmeans/month)](https://pepy.tech/project/fastkmeans) -->
20
21
 
21
22
  _A fast and efficient k-means implementation for PyTorch, with support for GPU and CPU._
@@ -96,3 +97,15 @@ Then, run the benchmark script:
96
97
  ```bash
97
98
  CUDA_VISIBLE_DEVICES=0 python speedbench.py --do-faiss --do-fastkmeans --do-fastkmeans-triton --do-evals
98
99
  ```
100
+
101
+ ### Citation
102
+
103
+ If you use fastmeans and want to/need to cite it in your work, please feel free to use the citation below:
104
+
105
+ ```bibtex
106
+ @misc{fastkmeans2025,
107
+ author = {Benjamin Clavié and Benjamin Warner},
108
+ title = {fastkmeans: Accelerated KMeans Clustering in PyTorch and Triton},
109
+ year = {2025},
110
+ howpublished = {\url{https://github.com/AnswerDotAI/fastkmeans/}}
111
+ }
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "fastkmeans"
3
- version = "0.4.0"
3
+ version = "0.5.0"
4
4
  description = "Add your description here"
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.8"
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/LICENSE RENAMED
File without changes
{fastkmeans-0.4.0 → fastkmeans-0.5.0}/setup.cfg RENAMED
File without changes