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.