A framework for simulating ecosystems in JAX.
Project description
EcoJAX
Research project for studying ecosystems of neural agents. In the context of my internship at FLOWERS.
Installation
Clone the repository and create a virtual environment. The python version used was 3.12.2.
git clone git@github.com:tboulet/EcoJAX.git
cd EcoJAX
python -m venv venv
source venv/bin/activate # on Windows, use `venv\Scripts\activate.bat`
Install JAX
On linux (note your CUDA version may var) :
pip install jax[cuda12_pip]==0.4.24 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
On Windows :
pip install jax[cpu]==0.4.24
Install the requirements
pip install -r requirements.txt
Install EvoJAX (for now)
cd evojax
pip install -e .
cd ..
Usage
To run any experiment, run the run.py
file with the desired configuration. We use Hydra as our configuration manager.
python run.py
To use a specific configuration, use the --config-name
argument.
python run.py --config-name=my_config
You can modify a specific argument thanks to Hydra's override system.
python run.py path.to.arg.in.config=new_value
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
ecojax-1.0.tar.gz
(37.4 kB
view hashes)
Built Distribution
ecojax-1.0-py3-none-any.whl
(43.7 kB
view hashes)