PyPI page
Home page
Author:
None
License:
MIT
Summary:
A unified deep learning model building library for PyTorch and JAX with shared dataclass-based configuration
Latest version:
0.2.4
Required dependencies:
einops
|
kornia
|
natsort
|
omegaconf
|
pytorch-lightning
|
pytorch-optimizer
|
schedulefree
|
torch
|
torchgeometry
|
torchvision
Optional dependencies:
codespell
|
distrax
|
einops
|
flax
|
ipython
|
jax
|
mkdocs
|
mkdocs-git-revision-date-localized-plugin
|
mkdocs-material
|
mkdocstrings
|
mypy
|
optax
|
pip
|
pre-commit
|
pytest
|
pytest-cov
|
ruff
|
types-pyserial
Downloads last day:
12
Downloads last week:
76
Downloads last month:
189