PyPI page
Home page
Author:
Mahdi Shafiei
License:
MIT License
Copyright (c) 2025 Mahdi Shafiei
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated docu...
Summary:
Datarax: A high-performance, NNX-based data pipeline framework for JAX
Latest version:
0.1.3
Required dependencies:
array-record
|
calibrax
|
flax
|
grain-nightly
|
jax
|
jax-metal
|
jaxlib
|
jaxtyping
|
numpy
|
nvidia-dali-cuda120
|
optax
|
orbax-checkpoint
|
protobuf
|
psutil
|
pyyaml
|
spdl
|
tensorflow
|
tensorflow-datasets
|
toml
|
tomli
|
tomli-w
Optional dependencies:
bandit
|
beartype
|
build
|
click
|
coverage
|
datasets
|
deadcode
|
deeplake
|
deptry
|
flake8
|
flake8-functions-names
|
google-cloud-aiplatform
|
google-cloud-storage
|
griffe
|
h5py
|
hippogriffe
|
import-linter
|
importlib-resources
|
interrogate
|
ipykernel
|
jax-dataloader
|
jupytext
|
keras
|
librosa
|
litdata
|
matplotlib
|
mkdocs
|
mkdocs-gen-files
|
mkdocs-include-exclude-files
|
mkdocs-ipynb
|
mkdocs-jupyter
|
mkdocs-literate-nav
|
mkdocs-material
|
mkdocstrings
|
mkdocstrings-python
|
mosaicml-streaming
|
pillow
|
pre-commit
|
pyarrow
|
pydeps
|
pylint
|
pymdown-extensions
|
pyright
|
pytest
|
pytest-asyncio
|
pytest-benchmark
|
pytest-cov
|
pytest-env
|
pytest-json-report
|
pytest-randomly
|
pytest-timeout
|
pytest-xdist
|
python-dotenv
|
radon
|
ray
|
rich
|
ruff
|
shellcheck-py
|
skypilot
|
soundfile
|
spdl
|
tabulate
|
tensorflow-datasets
|
torch
|
torchaudio
|
torchcrepe
|
torchvision
|
twine
|
vastai-sdk
|
vulture
|
webdataset
|
wemake-python-styleguide
|
xenon
Downloads last day:
90
Downloads last week:
1,257
Downloads last month:
2,332