InstallationΒΆ

JAX and Flax are required for installation (see instructions there).

Install TTAX from PyPi:

pip install ttax