FluxML/Zygote.jl

Relax! Flux is the ML library that doesn't make you tensor

An acausal modeling framework for automatically parallelized scientific machine learning (SciML) in Julia. A computer algebra system for integrated symbolics for physics-informed machine learning and automated transformations of differential equations

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

Efficiently computes derivatives of NumPy code.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

High-performance automatic differentiation of LLVM and MLIR.

Bayesian inference with probabilistic programming.

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Physics-Informed Neural Networks (PINN) Solvers of (Partial) Differential Equations for Scientific Machine Learning (SciML) accelerated simulation