shapels

Shape inference for PyTorch tensors directly inside VS Code. The extension bundles the shapels language server binary for Linux, macOS and Windows so users can get diagnostics without extra setup.

Features

• Reports tensor shape inference diagnostics as you type.
• Runs inference for tensor shapes, allowing for hovering over variables to inspect their shape.

Requirements

No external dependencies. The platform-specific server binary is shipped inside the extension (shapels-bin/<platform>/shapels).

Extension Settings

• shapels.serverPath: Override the bundled server binary path.

Release Notes

0.3.0

• Interactive tutorial at https://carrascomj.github.io/shapelsbook/
• Support for creation ops like tensor.zeros, tensor.ones and shape unrolling! This allows for not using jaxtyping at all if desired.
• Support for indexing.
• Support for comparison operators.
• Support for .to, and other operations like bool, int, float, etc.
• Support for randn, rand, detach and cuda.