Posts by Zekun Shi

1 results

Clear filters
  • SEPT. 9, 2025 / AI

    Beyond backpropagation: JAX's symbolic power unlocks new frontiers in scientific computing

    JAX, a framework known for large-scale AI model development, is proving to be a powerful tool in scientific computing, particularly for solving complex Partial Differential Equations (PDEs), now being leveraged by researchers to achieve significant speed-ups and memory reductions in solving high-order PDEs and demonstrating its potential to unlock new frontiers in scientific discovery.

    JAX's-Symbolic-Power-Blog-Meta