JeremieMelo / pytorch-onn
A PyTorch Library for Photonic AI Computing Model Training and Co-Design (NeurIPS'21)
Why this job: works in the areas we cover, but does not say which job it does
What it does
- Support pytorch training MZI-based ONNs. Support MZI-based Linear, Conv2d, BlockLinear, and BlockConv2d. Support weight, usv, phase modes and their conversion.
- Support phase quantization and non-ideality injection, including phase shifter gamma error, phase variations, and crosstalk.
- CUDA-accelerated batched MZI array decomposition and reconstruction for ultra-fast real/complex matrix mapping, which achieves 10-50X speedup over CPU-based unitary group parametrization. Francis (Triangle), Reck (Triangle), Clements (Recta
- Support pytorch training general frequency-domain ONNs (Gu+, FFT-ONN ASP-DAC 2020) (Gu+, FFT-ONN-v2 IEEE TCAD 2021) (Feng+, PSNN). Support FFT-ONN BlockLinear, and BlockConv2d. Support fft, hadamard, zero_bias, and trainable modes.
- Support multi-operand ring-based ONN architecture (Gu+, SqueezeLight DATE 2021). Support AllpassMORRCirculantLinear, AllpassMORRCirculantConv2d with built-in MORR nonlinearity.
Get started
pip install torchonn- Maintenance
- Inactive
- Last commit 10 months ago
- Stars
- 332
- On GitHub
- Licence
- MIT
- Allows business use
- Downloads for
- None listed
- No release files
- Latest version
- None yet
- No releases
- Built with
- Python
- 90% of the code
Is it looked after?
- Contributors
- Not stated
- Last 90 days
- Issues
- 0% closed
- 5 still open
- Pull requests
- 0 merged
- 1 waiting
- Releases
- None
- Never published
- No contributing guide
- No security policy
- No code of conduct
Similar repos
| Repo | Stars | Last commit | Licence | Downloads for |
|---|---|---|---|---|
| meta-pytorch / tnt | 1,723 | 3 days ago | Custom | No downloads |
| MatthewFilipovich / torchoptics | 179 | 7 months ago | MIT | No downloads |
| mosaicml / streaming | 1,571 | 9 days ago | Apache-2.0 | No downloads |
More: languages, topics and project record
Languages
- Python 90%
- Cuda 7%
- C++ 3%
- Shell <1%
Topics
- machine-learning
- neural-network
- optics
- pytorch
- optical-computing
- optical-neural-network
- pytorch-onn
Project record
- JobOther AI tools
- StatusListed
- Commits61
- Forks40
- Watching9
- Stars in 28 days+2
- CreatedJun 6, 2021
- First seen by usNov 7, 2026
- CheckedNov 8, 2026