Vibhatha Lakmal Abeykoon's repositories
jupyter_notebook_extensions
Initial Step
torchbenchmark
Internal Operation Overheads
AISC-Benchmarks-PyTorch
DNN training benchmarks in PyTorch
AllReduceBenchmark
AllReduce Benchmark
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
chainer
A flexible framework of neural networks for deep learning
DeepLearningExamples
Deep Learning Examples
DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
edl
Elastic Deep Learning using PaddlePaddle and Kubernetes
elastic
PyTorch elastic training
horovod
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
modin
Modin: Speed up your Pandas workflows by changing a single line of code
nccl-tests
NCCL Tests
petastorm
Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.
pip-packaging
Example Pip Package Tutorial [Following Standard Pip Docs]
pybindlib
Example PyBind Package
tensorpipe
A tensor-aware point-to-point communication primitive for machine learning
tfx
TFX is an end-to-end platform for deploying production ML pipelines
TimelineGraphTool
Drawing Timeline Graphs
torchgpipe
A GPipe implementation in PyTorch
twister2
Twister2 Big Data Hosting Environment: A composable framework for high-performance data analytics
twister2flow
Integrating Twister2 with external Computation Engines
UnoPytorch
Multi-tasking (drug response, cell line classification, etc.) Uno Implemented in PyTorch.
zyplots
ZyPlots