├── .gitignore ├── LICENSE ├── README.md ├── benchmarks ├── benchmark_fb.py └── benchmark_io.py ├── requirements.txt ├── setup.cfg ├── setup.py ├── strux ├── __init__.py ├── cky_crf.py ├── deptree.py ├── helpers.py ├── linearchain.py ├── semimarkov.py └── semirings.py └── tests ├── __init__.py ├── extensions.py └── test_jax.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # strux -------------------------------------------------------------------------------- /benchmarks/benchmark_fb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/benchmarks/benchmark_fb.py -------------------------------------------------------------------------------- /benchmarks/benchmark_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/benchmarks/benchmark_io.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | jax 2 | jaxlib 3 | -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/setup.py -------------------------------------------------------------------------------- /strux/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/__init__.py -------------------------------------------------------------------------------- /strux/cky_crf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/cky_crf.py -------------------------------------------------------------------------------- /strux/deptree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/deptree.py -------------------------------------------------------------------------------- /strux/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/helpers.py -------------------------------------------------------------------------------- /strux/linearchain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/linearchain.py -------------------------------------------------------------------------------- /strux/semimarkov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/semimarkov.py -------------------------------------------------------------------------------- /strux/semirings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/strux/semirings.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tests/extensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/tests/extensions.py -------------------------------------------------------------------------------- /tests/test_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/harvardnlp/strux/HEAD/tests/test_jax.py --------------------------------------------------------------------------------