{"library":"tokamax","install":[{"cmd":"pip install -U tokamax","imports":["import tokamax","import jax","import jax.numpy as jnp","tokamax.layer_norm","tokamax.dot_product_attention","tokamax.autotune","tokamax.standardize_function","tokamax.benchmark"]},{"cmd":"pip install git+https://github.com/openxla/tokamax.git","imports":[]}]}