ZhaoyueCheng / RankML

RankML Library in Jax / Keras

RankML

RankML Library in Jax and Keras

Setup

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

Install dependencies:

pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
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