Repository navigation

Docs cleaning + JAX implementation of the glm models - #52

Closed

ryan-odea wants to merge 15 commits into

CausalInference:mainfrom

Closed

Docs cleaning + JAX implementation of the glm models#52
ryan-odea wants to merge 15 commits into
CausalInference:mainfrom
ryan-odea:main

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
- Merge pull request #1 from ryan-odea/jax-logistic
  ryan-odea
  authored
  a397fdd
- pytree implementation working? + test
  ryan-odea
  committed
  c0e2d35
- Merge pull request #2 from ryan-odea/jax-logistic
  ryan-odea
  authored
  d3c0cb8