Open1

jaxtypingでNumPyやPyTorchのshapeを型安全にする