petastorm

Deep learning data loader

Enables training and evaluation of deep learning models from Apache Parquet datasets in various machine learning frameworks

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.

GitHub

2k stars
40 watching
284 forks
Language: Python
last commit: almost 3 years ago
Linked from 1 awesome list

deep-learningmachine-learningparquetparquet-filespyarrowpysparkpytorchsysmltensorflow

Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
open-mmlab/mmengineProvides a flexible and configurable framework for training deep learning models with PyTorch.1,196
uber/neuropodA unified interface to run deep learning models from multiple frameworks using C++ and Python.937
erotemic/netharnA PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments.40
eduardoleao052/js-pytorchA JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers.1,093
baguasys/baguaA framework for accelerating PyTorch deep learning training876
dmmiller612/sparktorchA PyTorch implementation on Apache Spark for distributed deep learning model training and inference.339
pyg-team/pytorch-frameA deep learning framework for handling heterogeneous tabular data with diverse column types582
zhanghang1989/pytorch-encodingA Python framework for building deep learning models with optimized encoding layers and batch normalization.2,044
graal-research/poutyneA PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities572
isht7/pytorch-deeplab-resnetA deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks.602
xxradon/igcv3-pytorchReimplements MobileNet-V2 and IGCV3 using PyTorch for efficient deep learning.19
ramon-oliveira/aorunA deep learning framework on top of PyTorch for building neural networks.61
rdspring1/pytorch_gbw_lmTrains a large-scale PyTorch language model on the 1-Billion Word dataset123
4uiiurz1/pytorch-res2netImplementations of deep learning architectures using PyTorch for image classification tasks on various datasets.112
dmarnerides/pydltA PyTorch-based toolbox for building and training deep learning models with ease.204