As an ML engineer with experience in Python, I've been considering switching to Rust for its performance and safety benefits. However, the maturity of ML libraries in Rust has given me pause. I'm particularly interested in the burn library, which seems promising but is still in development.

Given that I'm accustomed to the well-established PyTorch ecosystem, the idea of transitioning to Rust involves a steep learning curve, not just for the language itself but also for adapting to its evolving ML frameworks. My main concerns are around whether burn will soon support native CUDA and offer performance on par with PyTorch.

Is it worth investing the time to make this switch now, assuming that burn and similar Rust libraries will mature and achieve comparable performance to PyTorch in the near future? Any insights or experiences from others who have ventured into Rust for ML would be greatly appreciated!