salesforce/matchbox
A PyTorch library for writing deep learning code at the example level and executing it efficiently on minibatches.

Velocity · 7d
+0.2
★ / day
Trend
→steady
star history
Matchbox provides researchers with a MaskedBatch type that automatically tracks padding and masking for variable-size data. It includes a @batch decorator that rewrites Python control flow into SIMT-like forms with execution masking and synchronization primitives. The library offers convenience methods for common dynamic neural network operations, working seamlessly with both MaskedBatch and plain tensors.