#

jax2tf

Here is 1 public repository matching this topic...

wolfwdavid / jax-pinn

JAX/Flax physics-informed neural network with jax2tf export — benchmark JAX vs PyTorch vs TensorFlow

benchmark machine-learning scientific-computing flax structural-mechanics jax model-export physics-informed-neural-network neural-surrogate jax2tf

Updated Mar 21, 2026
Python

Improve this page

Add a description, image, and links to the jax2tf topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the jax2tf topic, visit your repo's landing page and select "manage topics."