Metadata-Version: 2.5
Name: aiaccel
Version: 2026.10.2
Summary: AIST Toolkit for Accelerating Machine Learning Research
Project-URL: documentation, https://aistairc.github.io/aiaccel/
Project-URL: repository, https://github.com/aistairc/aiaccel
Author-email: AIST <onishi-masaki@aist.go.jp>
License: MIT
License-File: LICENSE
License-File: LICENSE_HEADER
Classifier: License :: OSI Approved :: MIT License
Requires-Python: >=3.10
Requires-Dist: attrs
Requires-Dist: colorama
Requires-Dist: h5py
Requires-Dist: huggingface-hub
Requires-Dist: hydra-core
Requires-Dist: jinja2
Requires-Dist: lightning>=2.2.1
Requires-Dist: numpy
Requires-Dist: omegaconf
Requires-Dist: optuna>=4.5.0
Requires-Dist: questionary>=2.0
Requires-Dist: rich
Requires-Dist: scipy
Requires-Dist: simpleeval
Requires-Dist: tensorboard
Requires-Dist: torch>=2.2.0
Requires-Dist: typing-extensions
Description-Content-Type: text/markdown

<div align="center"><img src="https://raw.githubusercontent.com/aistairc/aiaccel/master/docs/image/logo_aiaccel.png" width="600"/></div>
<div align="center">
    <img src="https://img.shields.io/github/license/aistairc/aiaccel.svg" />
    <img src="https://img.shields.io/badge/Python-3.10-blue" />
    <a href="https://aiaccel.readthedocs.io/en/latest/index.html">
        <img src="https://app.readthedocs.org/projects/aiaccel/badge/?version=latest" />
    </a>
    <img src="https://github.com/aistairc/aiaccel/actions/workflows/ci.yaml/badge.svg" />
</div>

# AIST Toolkit for Accelerating Machine Learning Research

* **Research-Oriented**: designed to accelerate your research cycles written in Python
* **HPC Optimized**: intended to use in HPC clusters, including [AI Bridging Cloud Infrastructure (ABCI)](https://abci.ai/)
* **Highly Modular**: designed to let you pick up any part of aiaccel for your research project

# Key Features
* [PyTorch/Lightning Toolkit](https://aiaccel.readthedocs.io/en/latest/api_reference/torch.html): training toolkit for HPC clusters.
* [Hyperparameter Optimization (HPO)](https://aiaccel.readthedocs.io/en/latest/api_reference/hpo.html): ready-to-use HPO algorithms/tools.
* [OmegaConf Utilities](https://aiaccel.readthedocs.io/en/latest/api_reference/config.html): OmegaConf-based config utilities.


# Installation
```bash
pip install aiaccel
```
       
# Acknowledgement
* Part of this software was developed in a project commissioned by the New Energy and Industrial Technology Development Organization (NEDO).
* Part of this software was developed by using ABCI 3.0 provided by AIST and AIST Solutions.
* Part of this software was developed by using the TSUBAME4.0 supercomputer at Institute of Science Tokyo.
