https://github.com/danieldk/thinc-rust-ops
Non-official(!) Rust Thinc Ops, with SIMD-vectorized implementations of many ops. Because Ops are fun to implement.
https://github.com/danieldk/thinc-rust-ops
Last synced: 11 months ago
JSON representation
Non-official(!) Rust Thinc Ops, with SIMD-vectorized implementations of many ops. Because Ops are fun to implement.
- Host: GitHub
- URL: https://github.com/danieldk/thinc-rust-ops
- Owner: danieldk
- Created: 2022-06-24T12:44:53.000Z (about 4 years ago)
- Default Branch: main
- Last Pushed: 2022-09-07T08:31:55.000Z (almost 4 years ago)
- Last Synced: 2025-07-24T09:06:18.441Z (11 months ago)
- Language: Rust
- Homepage:
- Size: 114 KB
- Stars: 3
- Watchers: 3
- Forks: 1
- Open Issues: 6
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 🧪 thinc-rust-ops
This is an experimental, non-official, Python package that implements some Thinc
[ops](https://thinc.ai/docs/api-backends) using vectorized SIMD functions in Rust 🦀.
The following SIMD instruction sets are currently supported:
* Scalar
* ARM64:
- NEON
* x86_64:
- SSE2
- SSE4.1
- AVX
- AVX2 + FMA
## ⏳ Install
Make sure that you have a [Rust toolchain](https://rustup.rs) installed and then
install with `pip`:
```
python -m pip install git+https://github.com/danieldk/thinc-rust-ops#subdirectory=thinc-rust-ops
```
## 🚀 Quickstart
After installation, you can import `RustOps` and use it like any other `Ops` implementation:
```
>>> import numpy as np
>>> from rust_ops import RustOps
>>> ops = RustOps()
>>> ops.gelu(np.arange(-5., 5.))
array([-1.43552511e-06, -1.26744153e-04, -4.04990215e-03, -4.55001283e-02,
-1.58655251e-01, 0.00000000e+00, 8.41344749e-01, 1.95449987e+00,
2.99595010e+00, 3.99987326e+00])
```