PyPI page
Home page
Author:
None
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:
ARTIFEX: Architectures + Generative + JAX - A modular generative modeling library
Latest version:
0.1.0
Required dependencies:
blackjax
|
calibrax
|
dacite
|
datarax
|
datasets
|
distrax
|
flax
|
grain
|
graphviz
|
huggingface-hub
|
jax
|
jax-metal
|
jax-triton
|
jaxlib
|
jaxtyping
|
matplotlib
|
mlflow
|
mock
|
nglview
|
numpy
|
nvidia-cublas-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
|
optax
|
orbax-checkpoint
|
pandas
|
pre-commit
|
psutil
|
py3dmol
|
pyasn1
|
pydantic
|
pytest
|
pytest-benchmark
|
pytest-cov
|
pytest-xdist
|
python-dotenv
|
requests
|
scikit-learn
|
scipy
|
tqdm
|
trimesh
|
triton
|
typer
|
types-psutil
|
typing-extensions
|
wandb
Optional dependencies:
bandit
|
beartype
|
build
|
coverage
|
deadcode
|
deptry
|
flake8
|
flake8-functions-names
|
griffe
|
hippogriffe
|
import-linter
|
interrogate
|
ipykernel
|
jupyter
|
jupytext
|
mkdocs
|
mkdocs-include-exclude-files
|
mkdocs-ipynb
|
mkdocs-jupyter
|
mkdocs-material
|
mkdocstrings
|
mkdocstrings-python
|
nbconvert
|
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
|
ruff
|
tensorboard
|
twine
|
vulture
|
wemake-python-styleguide
Downloads last day:
16
Downloads last week:
136
Downloads last month:
1,249