Metadata-Version: 2.1
Name: torchmix
Version: 0.1.0rc8
Summary: Flexible components for transformers 🧩
Author: junhsss
Author-email: junhsssr@gmail.com
Requires-Python: >=3.8
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Dist: einops (>=0.6.0)
Requires-Dist: hydra-core (>=1.0.0)
Requires-Dist: hydra-zen (>=0.8.0)
Requires-Dist: jaxtyping (>=0.2.0)
Description-Content-Type: text/markdown

<h1 align="center">torchmix</h1>

<h3 align="center">The missing component library for PyTorch</h3>

<br />

`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.**

## Install

To use `torchmix`, you will need to have `torch` already installed on your environment.

```sh
pip install torchmix
```

## Documentation

To learn more, check out our [documentation](https://torchmix.vercel.app).

## Contributing

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.

## License

`torchmix` is licensed under the MIT License.

