Create your own GitHub profile
Sign up for your own profile on GitHub, the best place to host code, manage projects, and build software alongside 50 million developers.
Sign up
Pinned
209 contributions in the last year
Contribution activity
June 2020
- jaxpr typechecking: custom rules and errors
- de-duplicate constants when lowering to XLA
- summary statistics of jaxpr equations
- update jaxlib version in readme
- jaxpr pretty-print: wrap equation RHS when the LHS is long
- factor autodiff and vmap tests out from lax_test
- reuse input/output values when joining partially evaluated conditional branches
- jaxpr: drop unused values
- indexed conditionals with multiple branches