akihironitta
ML Software Engineer at @kumo-ai. Past: Rearch Engineer at @lightning-ai
@kumo-ai @pyg-teamLondon
Pinned Repositories
akihironitta
carettab
CaretTab - New Tab Page Replacement
generative-deep-learning
[WIP] PyTorch implementation of models in Generative Deep Learning book by David Foster.
gist
Collection of PyTorch examples, each within a single, self-contained Python file.
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
plwm
PLWM is a Python package, containing classes suitable for implementing a window manager. PLWM is also a window manager, using the PLWM package.
pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
pytorch_geometric
Graph Neural Network Library for PyTorch
akihironitta's Repositories
akihironitta/gist
Collection of PyTorch examples, each within a single, self-contained Python file.
akihironitta/akihironitta
akihironitta/carettab
CaretTab - New Tab Page Replacement
akihironitta/lightning-bolts
PyTorch Lightning Bolts is a toolbox of models, callbacks, and datasets for AI/ML researchers.
akihironitta/lightning-flash
Collection of tasks for fast prototyping, baselining, finetuning and solving problems with deep learning.
akihironitta/playground
A playground for CI/CD
akihironitta/pyg-benchmark
akihironitta/pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
akihironitta/akihironitta.github.io
akihironitta/metrics
Machine learning metrics for distributed, scalable PyTorch applications.
akihironitta/zdevito.github.io
akihironitta/lightning-benchmarks
Benchmarks on PyTorch Lightning to detect regressions in performance
akihironitta/actions
Another playground for CI/CD
akihironitta/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.
akihironitta/asv
Airspeed Velocity: A simple Python benchmarking tool with web-based reporting
akihironitta/azure-pipelines-vscode
VS Code extension for working with Azure Pipelines YAML files
akihironitta/data-structures-and-algorithms
拙著「問題解決力を鍛える!アルゴリズムとデータ構造」の補足資料。ソースコードと、章末問題への略解を掲載。
akihironitta/dotfiles
akihironitta/ecosystem-ci
Automate issue discovery for your projects against Lightning nightly and releases.
akihironitta/GitHubGraduation-2022
Join the GitHub Graduation Yearbook and "walk the stage" on June 11.
akihironitta/lightning-transformers
Flexible interface for high-performance research using SOTA Transformers leveraging Pytorch Lightning, Transformers, and Hydra.
akihironitta/optuna
A hyperparameter optimization framework
akihironitta/pyDeprecate
Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.
akihironitta/pyg-lib
Low-Level Graph Neural Network Operators for PyG
akihironitta/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
akihironitta/pytorch_geometric
Graph Neural Network Library for PyTorch
akihironitta/vision
Datasets, Transforms and Models specific to Computer Vision
akihironitta/x-stable-diffusion
Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention.
akihironitta/xla
Enabling PyTorch on Google TPU
akihironitta/xview3-reference
Reference data processing code and model for the xView3 prize challenge.