Skip to content

Commit 6439f5d

Browse files
authored
Update README.md
1 parent 038e52d commit 6439f5d

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

‎README.md‎

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,21 @@
44
[![Build Status](https://travis-ci.org/harvardnlp/pytorch-struct.svg?branch=master)](https://travis-ci.org/harvardnlp/pytorch-struct)
55
[![Coverage Status](https://coveralls.io/repos/github/harvardnlp/pytorch-struct/badge.svg?branch=master)](https://coveralls.io/github/harvardnlp/pytorch-struct?branch=master)
66

7-
<center><img src="https://github.com/harvardnlp/pytorch-struct/raw/master/download.png"></center>
7+
<p align="center">
8+
<img src="https://github.com/harvardnlp/pytorch-struct/raw/master/download.png">
9+
</p>
810

911

1012

11-
A library of tested, GPU implementations of core structured prediction algorithms for deep learning applications.
13+
A library of tested, GPU implementations of core structured prediction algorithms for deep learning applications.
14+
(or an implementation of <a href="https://www.cs.jhu.edu/~jason/papers/eisner.spnlp16.pdf">Inside-Outside and Forward-Backward Algorithms Are Just Backprop"<a/>)
15+
1216

1317
## Getting Started
1418

15-
> pip install .
19+
```
20+
pip install .
21+
```
1622

1723
```python
1824
import torch_struct
@@ -44,7 +50,7 @@ v, _ = torch.deptree_inside(scores, semiring=torch_struct.MaxSemiring)
4450

4551
## Library
4652

47-
Implemented:
53+
Current algorithms implemented:
4854

4955
* Linear Chain (CRF / HMM)
5056
* Semi-Markov (CRF / HSMM)
@@ -53,11 +59,11 @@ Implemented:
5359

5460
Design Strategy:
5561

56-
1) Minimal code, very small implementatations.
57-
2) All algorithms are implemented for GPU.
58-
3) Only implement forward pass, use gradients for marginals.
62+
1) Minimal implementatations. Most are 10 lines.
63+
2) Batched for GPU.
64+
3) Code can be ported to other backends
5965

60-
Semirings with Gradients
66+
Semirings:
6167

6268
* Log Marginals
6369
* Max and MAP computation

0 commit comments

Comments
 (0)