Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
__init__.py		__init__.py
install.sh		install.sh
optimize.py		optimize.py

README.md

JAX Linear Model

In this example, we show how to optimize JAX linear model. We recommend running this example in NVIDIA NGC TensorFlow2 container.

Set up the environment

. ./install.sh

Run optimization

./optimize.py