Masaki Kozuki's repositories
lightning-thunder
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
Fuser
A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
text
Models, data loaders and abstractions for language processing, powered by PyTorch
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
OpenNMT-py
Open Source Neural Machine Translation in PyTorch
ucx
Unified Communication X (mailing list - https://elist.ornl.gov/mailman/listinfo/ucx-group)
ucc
Unified Communication Collectives Library
python_example
Example pybind11 module built with a Python-based build system
pybind11
Seamless operability between C++11 and Python
NeMo
NeMo: a toolkit for conversational AI
crcrpar
Hi
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
DeepLearningExamples
Deep Learning Examples
awesome-explainable-graph-reasoning
A collection of research papers and software related to explainability in graph machine learning.
pytorch.github.io
The website for PyTorch
pytorch-tutorials
PyTorch tutorials.
optuna
A hyperparameter optimization framework
vision
Datasets, Transforms and Models specific to Computer Vision
sphinx-gallery
Sphinx extension for automatic generation of an example gallery
optuna-mirror
I mirror https://github.com/optuna/optuna for my personal purpose
cuda-samples
Samples for CUDA Developers which demonstrates features in CUDA Toolkit
m-pw-pw
mirror proxy-wasm/proxy-wasm
bazel-hermetic-python
Example of building custom python interpreter in bazel with pip support.