There's some poc xla bindings for rust. It would be really cool to integrate those: https://github.com/LaurentMazare/xla-rs into dfdx to allow for tpus and other accelerators in addition to cuda.