Akihiro Nitta's repositories
lightning-bolts
PyTorch Lightning Bolts is a toolbox of models, callbacks, and datasets for AI/ML researchers.
lightning-flash
Collection of tasks for fast prototyping, baselining, finetuning and solving problems with deep learning.
playground
A playground for CI/CD
pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
lightning-benchmarks
Benchmarks on PyTorch Lightning to detect regressions in performance
AITemplate
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
azure-pipelines-vscode
VS Code extension for working with Azure Pipelines YAML files
data-structures-and-algorithms
拙著「問題解決力を鍛える!アルゴリズムとデータ構造」の補足資料。ソースコードと、章末問題への略解を掲載。
ecosystem-ci
Automate issue discovery for your projects against Lightning nightly and releases.
GitHubGraduation-2022
Join the GitHub Graduation Yearbook and "walk the stage" on June 11.
lightning-transformers
Flexible interface for high-performance research using SOTA Transformers leveraging Pytorch Lightning, Transformers, and Hydra.
pyDeprecate
Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.
pyg-lib
Low-Level Graph Neural Network Operators for PyG
pytorch_geometric
Graph Neural Network Library for PyTorch
relbench
RelBench: Relational Deep Learning Benchmark
x-stable-diffusion
Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention.
xview3-reference
Reference data processing code and model for the xView3 prize challenge.