torchmix
is a collection of pre-made PyTorch components designed to simplify model development process, primarily for transformers. Our goal is to enable easy adoption of cutting-edge components with minimal code and maximum scalability.
Note: torchmix
is a prototype that is currently in development and has not been tested for production use. The API may change at any time.
To use torchmix
, you will need to have torch
already installed on your environment.
pip install torchmix
To learn more, check out our documentation.
If you have ideas for new components or ways to enhance the library, feel free to open an issue or start a discussion. We welcome all forms of feedback, including criticism and suggestions for significant design changes.
torchmix
is licensed under the MIT License.