Repository navigation

Jax logistic - #53

Merged

ryan-odea merged 14 commits into

CausalInference:mainfrom

ryan-odea:jax-logistic

Jun 8, 2026

Merged

Jax logistic#53
ryan-odea merged 14 commits into
CausalInference:mainfrom
ryan-odea:jax-logistic

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