From Julia to Rust: a differentiable tensor stack for scientific computing
12 days ago
- #Scientific Computing
- #Differentiable Programming
- #Rust
- 由于Julia在大型代码库中存在运行时问题，因此从Julia转向Rust进行科学计算。
- tenferro-rs 是一个原生Rust密集张量库，支持自动微分、einsum运算、快速傅里叶变换（FFT）以及显式的CPU/CUDA后端。
- Rust的所有权系统和类型机制在编译期增强了代码正确性，使其特别适合由AI生成的代码。
- 其设计将操作族、自动微分规则和后端实现分离为独立的crate，以提高可扩展性。
- 支持PyTorch风格的即时自动微分和JAX风格的追踪变换，以处理动态形状计算。
- 通过有限差分验证、性能基准测试和设计文档来确保代码的正确性。