Jax logistic - #53

Merged

ryan-odea merged 14 commits into

CausalInference:mainfrom

ryan-odea:jax-logistic

Jun 8, 2026

Commits on Jun 8, 2026

add jax to the toml
ryan-odea
committed
5560634
clean up some dispatch
ryan-odea
committed
0665436
glum fit summary2 -> summary. Maybe can be hidden in main class
ryan-odea
committed
b5ab7ca
add jax to opts
ryan-odea
committed
46359a1
removed explicit typing (should be fine with the sphinx autodoc)
ryan-odea
committed
096f470
remove defined typing - let sphinx infer from type hints
ryan-odea
committed
64ca5fc
multinomial logistic via ADAM
ryan-odea
committed
3bc677b
update the glum test
ryan-odea
committed
57b9ca8
make a dev optional
ryan-odea
committed
c5b35ca
clone _glum_fit but with jax
ryan-odea
committed
cd4d5a7
tests
ryan-odea
committed
e0c7973
Update pyproject.toml
ryan-odea
committed
f2e0658
pytree implementation working? + test
ryan-odea
committed
c0e2d35
update to install devel options
ryan-odea
committed
b855e59