Installation¶

Requirements¶

# Standard installation
pip install myriad-jax

# With generic GPU dependencies (checks for nvidia-related packages)
pip install "myriad-jax[gpu]"

Clone the repository:

git clone https://github.com/robinhenry/myriad.git
cd myriad

Install dependencies with Poetry:

poetry install

Or with pip:

pip install -e .

poetry install --with gpu

Or:

pip install "jax[cuda12]"

JAX Metal support is experimental. Use CPU for now.

python -c "import myriad; print('Myriad installed successfully')"

Run tests:

python -m pytest