#pytorch #screenshot #wrapper #interface #ergonomics #ergonomic #tch

condor

Condor is an ergonomic wrapper around the tch library for Rust

7 releases

0.1.6 Oct 18, 2021
0.1.5 Oct 18, 2021
0.1.4 Aug 3, 2021
0.1.3 Jul 26, 2021

#456 in Machine learning

MIT/Apache

2MB
2.5K SLoC

Condor

Screenshot

Condor aims to be an ergonomic PyTorch interface in Rust.

CI Status Current Crates.io Version

TODO

  • Implement linear layers
  • Implement basic transformer layers
  • Implement sequential
  • Implement decoder and seq2seq transformer
  • Implement RNN layers
  • Implement utilities
  • Implement more exotic architectures
  • Write more detailed tests (check tensor data)

Dependencies

~8–16MB
~320K SLoC