Rust Burn is a new deep learning framework built entirely in Rust. It aims to balance flexibility, performance, and ease of use for researchers, ML engineers, and developers. The framework offers features such as a flexible computational graph, thread-safe data structures, intuitive abstractions, and support for both CPU and GPU implementations. Rust Burn can be installed by setting up Rust and using the Cargo package manager. Example projects and pre-trained models are available for exploration and fine-tuning.
Sort: