HomeAI Tutorial

JAX-NSL

Public

Comprehensive JAX implementation of neural networks and scientific computing. Features distributed training, physics-informed networks, custom autodiff, and advanced optimization. Production-ready code with numerical stability, multi-device parallelism, and research-grade implementations.

Creat2025-09-02T01:08:42
Update2025-09-03T09:43:28
1
Stars
0
Stars Increase