Fix dependency pins and enable static arguments#27

acmiyaguchi wants to merge 3 commits intozombie-einstein:mainzombie-einstein/jaxpr-viz:mainfrom

acmiyaguchi:mainacmiyaguchi/jaxpr-viz:mainCopy head branch name to clipboard