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:
Scientific machine learning for JAX/Flax NNX: neural operators, physics-informed networks, E(3)-equivariant atomistic potentials, differentiable DFT, SINDy equation discovery and uncertainty quantification, with PDEBench benchmarking
Latest version:
0.2.10
Required dependencies:
avitai-artifex
|
beartype
|
blackjax
|
calibrax
|
datarax
|
diffrax
|
flax
|
jax
|
jaxtyping
|
matplotlib
|
numpy
|
optax
|
optimistix
|
orbax-checkpoint
|
pyyaml
|
scipy
|
substrax
|
tensorly
|
tfp-nightly
|
tqdm
Optional dependencies:
aiosqlite
|
alembic
|
ase
|
asyncpg
|
bandit
|
basis-set-exchange
|
beartype
|
build
|
coverage
|
distrax
|
exponax
|
fastapi
|
grain
|
greenlet
|
griffe
|
h5py
|
hippogriffe
|
hydra-core
|
hypothesis
|
import-linter
|
interrogate
|
ipykernel
|
jax
|
jax-metal
|
jupytext
|
meshio
|
mkdocs
|
mkdocs-include-exclude-files
|
mkdocs-ipynb
|
mkdocs-jupyter
|
mkdocs-material
|
mkdocstrings
|
mkdocstrings-python
|
netcdf4
|
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
|
pandas
|
pgcli
|
pip-audit
|
pre-commit
|
prometheus-client
|
psycopg2-binary
|
pydeps
|
pydoclint
|
pymdown-extensions
|
pyright
|
pyscf
|
pytest
|
pytest-asyncio
|
pytest-benchmark
|
pytest-cov
|
pytest-env
|
pytest-json-report
|
pytest-randomly
|
pytest-split
|
pytest-timeout
|
pytest-xdist
|
python-dotenv
|
qcelemental
|
qcengine
|
radon
|
redis
|
rich
|
ruff
|
sqlalchemy
|
substrax
|
tensorboard-plugin-profile
|
tensorflow
|
tf-keras
|
torch
|
twine
|
uvicorn
Downloads last day:
4
Downloads last week:
46
Downloads last month:
4,320
7-day smoothing