{"library":"haliax","install":[{"cmd":"pip install haliax","imports":["from haliax import NamedArray","from haliax import Axis","from haliax import product","import haliax.nn as nn","from haliax.nn import Linear"]},{"cmd":"pip install haliax jax[cuda12_pip] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html","imports":[]}]}