jax

96e37ced - Merge pull request #613 from hawkinsp/xla