Major release modernizing the TensorFlow/CUDA stack and build system.
What's new
- TensorFlow 2.19 + Keras 3: Replaces TF 2.15. Fixes GPU underutilization issues (#65, #73) by replacing the broken
tf.compat.v1GPU configuration with native TF2 APIs. Model serialization updated from JSON+H5 to Keras 3.kerasformat. - pixi for dependency management: Single
pixi installhandles Python, TensorFlow, CUDA, cuDNN, and all dependencies. No more manual conda/pip setup. Supports GPU (default) and CPU-only (cpu) environments. - Forward-compatible CUDA via
cuda-compat: TF 2.19 (CUDA 12.8) works on older NVIDIA drivers (e.g. 535) without driver upgrades. - Unit tests + CI: 63 tests covering core utility functions, data preprocessing, and simulation methods. GitHub Actions runs tests on every push and PR.
- Python 3.10-3.12 support: Fixes
random.seed()with numpy integers (Python 3.12+),NearestNeighborsAPI change (scikit-learn), Keras 3model.fit()andplotResultscompatibility.
Installation
git clone https://github.com/kr-colab/ReLERNN.git
cd ReLERNN
pixi installSee the README for details.
Breaking changes
setup.pyandrequirements.txtremoved in favor ofpyproject.tomlandpixi.toml- Trained models are now saved as
.kerasfiles (notmodel.json+weights.h5). Models trained with v1.x cannot be loaded directly. - Requires Python >= 3.10