Installation

Stable version

The latest stable release of tnkm is available on PyPI:

pip install tnkm

Verify installation

You can verify that TNKM is installed correctly by checking the version:

python -c "import tnkm; print(tnkm.__version__)"

GPU/TPU support

TNKM is built on JAX and supports hardware acceleration (GPU/TPU).

To enable this, install JAX with the appropriate backend for your system. Please refer to the official JAX installation guide. for detailed instructions.

Note

Installing tnkm via pip will install the CPU version of JAX by default.

Development version

Warning

The development version may be unstable and include experimental features.

Clone the repository:

git clone https://github.com/AlbMLpy/tnkm.git
cd tnkm

and install locally:

pip install -e .

(Optional) If you use Poetry for development:

poetry install