PyPI 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:
Generative modeling for JAX/Flax NNX: VAEs, GANs, diffusion, normalizing flows, energy-based, autoregressive and geometric models across image, text, audio, molecular, protein, tabular and time-series modalities, with a shared Trainer and evaluation through calibrax
Latest version:
0.1.15
Required dependencies:
blackjax
|
calibrax
|
dacite
|
datarax
|
distrax
|
flax
|
grain
|
jax
|
jaxtyping
|
matplotlib
|
numpy
|
optax
|
orbax-checkpoint
|
psutil
|
pyasn1
|
pydantic
|
python-dotenv
|
requests
|
scikit-learn
|
scipy
|
substrax
|
tqdm
|
typing-extensions
Optional dependencies:
bandit
|
beartype
|
build
|
coverage
|
datasets
|
deadcode
|
deptry
|
detect-secrets
|
flake8
|
flake8-functions-names
|
graphviz
|
griffe
|
hippogriffe
|
huggingface-hub
|
hypothesis
|
import-linter
|
interrogate
|
ipykernel
|
jax
|
jax-metal
|
jaxlib
|
jupyter
|
jupytext
|
mkdocs
|
mkdocs-include-exclude-files
|
mkdocs-ipynb
|
mkdocs-jupyter
|
mkdocs-material
|
mkdocstrings
|
mkdocstrings-python
|
mlflow
|
mock
|
nbconvert
|
nvidia-cublas-cu12
|
nvidia-cuda-cccl-cu12
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvcc-cu12
|
nvidia-cuda-nvrtc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
|
nvidia-nvshmem-cu12
|
pip-audit
|
pydeps
|
pydoclint
|
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
|
ruff
|
substrax
|
tensorboard
|
trimesh
|
twine
|
typer
|
vulture
|
wandb
|
wemake-python-styleguide
|
xenon
Downloads last day:
17
Downloads last week:
279
Downloads last month:
8,107
7-day smoothing