tch-rs

PyTorch wrapper

Rust bindings for PyTorch's C++ API to facilitate machine learning and neural network development.

Rust bindings for the C++ api of PyTorch.

GitHub

4k stars
52 watching
347 forks
Language: Rust
last commit: about 1 month ago
Linked from 3 awesome lists

deep-learningmachine-learningneural-networkpytorchrust

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
prabhuomkar/pytorch-cpp A C++ implementation of PyTorch tutorials 1,978
pytorch/extension-cpp Enables the creation of custom C++ extensions with CUDA support in PyTorch 1,031
wizaron/pytorch-cpp-inference A repository providing tools and examples to serve PyTorch models as C++ inference applications. 226
pytorch/serve A tool for serving and scaling PyTorch models in production environments 4,238
tomassedovic/tcod-rs Rust bindings for a roguelike toolkit with C-like functionality and portable graphics support. 228
mozilla/cbindgen Generates C/C++ bindings from Rust code to facilitate interoperability with other languages and systems. 2,440
rust-lang/rust-bindgen Automates the process of creating Rust interfaces to C and C++ libraries. 4,531
amdegroot/ssd.pytorch An implementation of a deep learning-based object detection system in PyTorch. 5,160
dgrunwald/rust-cpython Rust bindings for the Python interpreter, allowing Rust code to interact with and use Python libraries and scripts. 1,815
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,523
slabban/ros2_pytorch_cuda Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment. 7
libpnet/libpnet A low-level networking library written in Rust for cross-platform development of transport protocols and network utilities. 2,325
libp2p/rust-libp2p An implementation of a peer-to-peer networking stack in Rust 4,664
pyo3/pyo3 Rust bindings for the Python interpreter 12,615
longcw/roialign.pytorch Implementation of RoIAlign and crop_and_resize functions for PyTorch 554