diff --git a/benchmarks/README.md b/benchmarks/README.md index 43e7368..ce3fa84 100644 --- a/benchmarks/README.md +++ b/benchmarks/README.md @@ -16,3 +16,23 @@ To reproduce the results in the paper, you can run ```sh python -m benchmarks.built-in.run_all ``` + +## Legacy Scripts + +> [!WARNING] +> These scripts are used to benchmark `jax-relax<0.2.0>=0.1.0`. +> + +To run these scripts, install the dependencies as: +```sh +pip install "jax-relax[dev]<0.2.0" +``` + +Next, run the script as: + +```sh +# run large dataset +python -m benchmarks.legacy.benchmark_large_dataset +# run scalability test +python -m benchmarks.legacy.benchmark_scalability +``` diff --git a/benchmarks/benchmark_large_dataset.py b/benchmarks/legacy/benchmark_large_dataset.py similarity index 100% rename from benchmarks/benchmark_large_dataset.py rename to benchmarks/legacy/benchmark_large_dataset.py diff --git a/benchmarks/benchmark_scalability.py b/benchmarks/legacy/benchmark_scalability.py similarity index 100% rename from benchmarks/benchmark_scalability.py rename to benchmarks/legacy/benchmark_scalability.py