PyPI page
Home page
Author:
None
License:
MIT License
Copyright (c) 2026 Mahdi Shafiei
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated docu...
Summary:
Calibrax: Unified benchmarking framework for the JAX scientific ML ecosystem
Latest version:
0.1.1
Required dependencies:
click
|
flax
|
jax
|
jaxlib
|
jaxtyping
|
numpy
|
psutil
|
typing-extensions
Optional dependencies:
beartype
|
build
|
codecarbon
|
coverage
|
griffe
|
hippogriffe
|
interrogate
|
ipykernel
|
jax-cuda12-pjrt
|
jax-cuda12-plugin
|
jax-metal
|
jupytext
|
matplotlib
|
mkdocs
|
mkdocs-include-exclude-files
|
mkdocs-ipynb
|
mkdocs-jupyter
|
mkdocs-material
|
mkdocstrings
|
mkdocstrings-python
|
mlflow
|
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-curand-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
|
nvidia-nvtx-cu12
|
pre-commit
|
pydeps
|
pygments
|
pymdown-extensions
|
pynvml
|
pyright
|
pytest
|
pytest-asyncio
|
pytest-benchmark
|
pytest-codspeed
|
pytest-cov
|
pytest-env
|
pytest-json-report
|
pytest-randomly
|
pytest-timeout
|
pytest-xdist
|
radon
|
ruff
|
ruptures
|
scikit-learn
|
scipy
|
twine
|
wandb
Downloads last day:
18
Downloads last week:
1,068
Downloads last month:
1,933