README.md

RankML

RankML Library for TPU in Keras and Flax

Setup

To set up the environment for this project, follow these steps:

Install dependencies:

pip install -r requirements.txt

Running the code

After setting up the environment, you can run the training script using the following command:

python3 train.py