impetus 0.1.1

An awesome tool/library benchmark LLM performance on all kinds of hardware!

Impetus

An awesome tool/library benchmark LLM performance on all kinds of hardware!
Explore the docs »

View Demo · Report Bug · Request Feature

Table of Contents

1. About The Project
   • Built With
2. Getting Started
   • Prerequisites
   • Installation
3. Usage
4. Roadmap
5. Contributing
6. License
7. Contact
8. Acknowledgments

About The Project

(back to top)

Getting Started

Installation

Installation with pip is simple as follows:

pip install impetus

Usage

For more examples, please refer to the Documentation

Developing locally

This project use poetry for dependency management, and is required for running impetus locally.

1. Clone the repo

   git clone https://github.com/dhmnr/impetus.git
   

2. Install Dependencies

   poetry install
   

3. Run the command as follows

   poetry shell run impetus --help
   

(back to top)

Roadmap

• Support Quantization
• Multiple GPU support with accelerate
• GPU/CPU usage information
• Flops/MFU information
• Detailed breakdown, | self-attention | FFN | lm_head
• Support BERT and VLMs and others
• Support different backends

Minor Fixes

• Support different datasets

See the open issues for a full list of proposed features (and known issues).

(back to top)

Contributing

Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.

If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement". Don't forget to give the project a star! Thanks again!

1. Fork the Project
2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
3. Commit your Changes (git commit -m 'Add some AmazingFeature')
4. Push to the Branch (git push origin feature/AmazingFeature)
5. Open a Pull Request

Top contributors:

(back to top)

License

Distributed under the MIT License. See LICENSE.txt for more information.

(back to top)

Contact

Project Link: https://github.com/dhmnr/impetus

(back to top)

Acknowledgments

Use this space to list resources you find helpful and would like to give credit to. I've included a few of my favorites to kick things off!

• Choose an Open Source License
• Img Shields
• GitHub Pages

(back to top)