Skip to content

Commit e24a25a

Browse files
committed
add readme
1 parent 9180474 commit e24a25a

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

README.md

+14
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
1+
### Introduction
2+
3+
This project is to speed up various ML models (e.g. topic modeling, word embedding, etc) by CUDA. It would be nice to think of it as [gensim](https://github.com/RaRe-Technologies/gensim)'s GPU version project. As a starting step, I implemented the most widely used word embedding model, the [word2vec](https://arxiv.org/pdf/1301.3781.pdf) model, and the most representative topic model, the [LDA (Latent Dirichlet Allocation)](https://www.jmlr.org/papers/volume3/blei03a/blei03a.pdf) model.
4+
15
### How to install
26

7+
- install from source
38

49
```shell
510
# clone repo and submodules
@@ -14,3 +19,12 @@ python -m grpc_tools.protoc --python_out cusim/ --proto_path cusim/proto/ config
1419
# install
1520
python setup.py install
1621
```
22+
23+
- pip installation will be available soon
24+
25+
### How to use
26+
27+
- `examples/example1.py` and `examples/README.md` will be very helpful to understand the usage.
28+
- paremeter description can be seen in `cusim/proto/config.proto`
29+
30+
### Performance

0 commit comments

Comments
 (0)