Metadata-Version: 2.4
Name: rbyte
Version: 0.39.0
Summary: Multimodal PyTorch dataset library
Author: Evgenii Gorchakov
Author-email: Evgenii Gorchakov <evgenii@yaak.ai>
License-File: LICENSE.txt
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Dist: cachetools>=7.1.4
Requires-Dist: checkedframe>=0.1.0
Requires-Dist: duckdb>=1.5.4
Requires-Dist: hydra-core>=1.3.3
Requires-Dist: more-itertools>=11.1.0
Requires-Dist: numpy
Requires-Dist: optree>=0.19.1
Requires-Dist: pipefunc>=0.93.1
Requires-Dist: polars>=1.41.2
Requires-Dist: pydantic>=2.13.4
Requires-Dist: structlog>=26.1.0
Requires-Dist: tensordict>=0.13.0
Requires-Dist: torch
Requires-Dist: torchdata>=0.11.0
Requires-Dist: tqdm>=4.67.1
Requires-Dist: xxhash>=3.7.0
Requires-Dist: h5py>=3.14.0 ; extra == 'hdf5'
Requires-Dist: simplejpeg>=1.9.0 ; extra == 'jpeg'
Requires-Dist: mcap>=1.3.1 ; extra == 'mcap'
Requires-Dist: mcap-protobuf-support>=0.5.4 ; extra == 'mcap'
Requires-Dist: ptars>=0.0.17 ; extra == 'mcap'
Requires-Dist: protobuf>=7.35.1,<8 ; extra == 'mcap'
Requires-Dist: torchcodec>=0.14.0 ; extra == 'video'
Requires-Dist: einops>=0.8.2 ; extra == 'visualize'
Requires-Dist: rerun-sdk[notebook]>=0.33.1 ; extra == 'visualize'
Requires-Dist: m2df ; extra == 'yaak'
Requires-Dist: ymmv ; extra == 'yaak'
Maintainer: Evgenii Gorchakov
Maintainer-email: Evgenii Gorchakov <evgenii@yaak.ai>
Requires-Python: >=3.12, <3.13
Project-URL: repo, https://github.com/yaak-ai/rbyte
Provides-Extra: hdf5
Provides-Extra: jpeg
Provides-Extra: mcap
Provides-Extra: video
Provides-Extra: visualize
Provides-Extra: yaak
Description-Content-Type: text/markdown

<p align="center">
 <a href="https://www.yaak.ai/open-source/dev-tools">
  <img alt="banner" src="https://github.com/user-attachments/assets/707ab3ae-73d5-459f-82c5-888323673adb">
 </a>
</p>

<p align="center">
 <a href="https://deepwiki.com/yaak-ai/rbyte"><img src="https://deepwiki.com/badge.svg" alt="Ask DeepWiki"></a>
 <img src="https://github.com/yaak-ai/rbyte/actions/workflows/ci.yaml/badge.svg">
 <img src="https://img.shields.io/github/license/yaak-ai/rbyte.svg?color=green"></a>
</p>

`rbyte` implements a [PyTorch](https://pytorch.org) [`Dataset`](https://pytorch.org/tutorials/beginner/basics/data_tutorial.html) with [`TensorClass`](https://docs.pytorch.org/tensordict/stable/reference/generated/tensordict.TensorClass.html)-based samples built from multimodal data.

## Installation

```bash
uv add rbyte [--extra <EXTRA>]
```

See `pyproject.toml` for available extras.

## Examples

```bash
git clone https://github.com/yaak-ai/rbyte && cd rbyte
nix develop # alternatively, install `uv`, `just`, `ytt`
just notebook examples/nuscenes.ipynb
```

## Development

```bash
git clone https://github.com/yaak-ai/rbyte && cd rbyte
nix develop
just setup
```
