danielward27
commented
1 year ago - Adds Stack and Concatenate
- Corrections and improvements to type hints and documentation
- Fixes bug I introduced in the variational inference code which led to the gradient being taken with respect to the wrong argument (the PRNGKey array)