Jax

entities · 2 notes linked

Related: Weights And Biases · Physics Informed Neural Networks · Scientific Machine Learning · Neural Tangent Kernel · Google · Numpy · Automatic Differentiation · Just In Time Compilation

Notes

• GitHub - PredictiveIntelligenceLab/jaxpi — JAX library implementing physics-informed neural networks with benchmarks
• JAX: Is This Google's NumPy killer? — JAX library combining NumPy API with autodiff and GPU/TPU acceleration