search

Modalities / modalities

Modalities, a PyTorch-native framework for distributed and reproducible foundation model training.
MIT License
63 stars 8 forks source link

Add Missing Type Annotations To Mamba Related Code #227

Open mali-git opened 3 months ago

mali-git commented 3 months ago

Feature request

Add type annotations to MAMBA related code: https://github.com/Modalities/modalities/blob/10c6022a6ac315d70180593185a0be6204abdc36/src/modalities/models/mamba/ops/triton/layernorm.py#L18-L614

Motivation

Improves code readability and quality.