torchax

fix the type conversion in checkpoint (#19)

Bugfix: torch.full(..., 0) should be int tensor (#18)

Use jupytext for notebooks (#9)

Add padding_idx gradient backprop stopping for Embeddings (#11)

how-it-works update (#13)