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)
• Highly Modular: designed to let you pick up any part of aiaccel for your research project

Key Features

• PyTorch/Lightning Toolkit: training toolkit for HPC clusters.
• Hyperparameter Optimization (HPO): ready-to-use HPO algorithms/tools.
• OmegaConf Utilities: OmegaConf-based config utilities.

Installation

pip install git+https://github.com/aistairc/aiaccel.git

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.