Burn Torch Backend

Burn Torch backend

This crate provides a Torch backend for Burn utilizing the tch-rs crate, which offers a Rust interface to the PyTorch C++ API.

The backend supports CPU (multithreaded), CUDA (multiple GPUs), and MPS devices (MacOS).

Usage Example

```rust

[cfg(feature = "tch-gpu")]

mod tchgpu { use burnautodiff::ADBackendDecorator; use burn_tch::{TchBackend, TchDevice}; use mnist::training;

pub fn run() {
    #[cfg(not(target_os = "macos"))]
    let device = TchDevice::Cuda(0);
    #[cfg(target_os = "macos")]
    let device = TchDevice::Mps;

    training::run::<ADBackendDecorator<TchBackend<f32>>>(device);
}

}

[cfg(feature = "tch-cpu")]

mod tchcpu { use burnautodiff::ADBackendDecorator; use burn_tch::{TchBackend, TchDevice}; use mnist::training;

pub fn run() {
    let device = TchDevice::Cpu;
    training::run::<ADBackendDecorator<TchBackend<f32>>>(device);
}

} ```