issues
search
LouisDesdoigts
/
zodiax
Object-oriented Jax framework extending Equinox for scientific programming
https://louisdesdoigts.github.io/zodiax/
BSD 3-Clause "New" or "Revised" License
11
stars
1
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
chi2 likelihood != normal
#37
benjaminpope
opened
3 months ago
0
Create utils.py
#36
benjaminpope
opened
6 months ago
0
Gradient of float returns None
#35
maxecharles
opened
10 months ago
0
Don't default to adam on leaves to prevent potential parameter tracking by optax objects
#34
LouisDesdoigts
opened
10 months ago
0
Enable gradients through filter function wrt floats
#33
LouisDesdoigts
opened
1 year ago
0
spelling of covariance
#32
benjaminpope
closed
1 year ago
1
Test and Implement optimised `filter_jit`
#31
LouisDesdoigts
opened
1 year ago
0
Typo in `bayes.covaraince_entropy` -> `covariance`
#30
LouisDesdoigts
opened
1 year ago
0
Enable the setting of static fields?
#29
LouisDesdoigts
opened
1 year ago
1
Nan covariance matrix when parameters have array followed by float
#28
LouisDesdoigts
opened
1 year ago
1
Setup rework
#27
LouisDesdoigts
closed
1 year ago
0
Path API
#26
ASEM000
opened
1 year ago
14
Add `tree.set_array` call to `filter_grad` etc to ensure gradient outputs
#25
LouisDesdoigts
opened
1 year ago
0
V0.4
#24
LouisDesdoigts
closed
1 year ago
0
add .divide method
#23
LouisDesdoigts
closed
1 year ago
1
Python requirement testing
#22
LouisDesdoigts
closed
1 year ago
0
Add default `.zdx` extension to serialise
#21
LouisDesdoigts
closed
1 year ago
1
Update README.md
#20
benjaminpope
closed
1 year ago
0
Update usage.md
#19
benjaminpope
closed
1 year ago
0
Package Restructure, Docs overhaul
#18
LouisDesdoigts
closed
1 year ago
1
Move the Bayesian module into Zodiax
#17
LouisDesdoigts
closed
1 year ago
1
Swap input order to `serialise`
#16
LouisDesdoigts
closed
1 year ago
0
Serialisation
#15
LouisDesdoigts
closed
1 year ago
0
Add default filter to `filter_grad`
#14
LouisDesdoigts
closed
1 year ago
1
Create an `optimisation` module
#13
LouisDesdoigts
closed
1 year ago
3
Add an `__eq__` method that calls `equinox.tree_equal` under the hood
#12
LouisDesdoigts
closed
1 year ago
1
Remove import from `abc.ABC` as `eqx.Module` is already abstract
#11
LouisDesdoigts
closed
1 year ago
0
Develop `filter` module to replace `equinox` filter functions that supports the `filter_spec` synatx
#10
LouisDesdoigts
closed
1 year ago
0
Investigate usage of jax `KeyPaths`
#9
LouisDesdoigts
opened
1 year ago
0
Fix large font in ExtendedBase tutorial
#8
LouisDesdoigts
closed
1 year ago
0
Equinox wrappers
#7
LouisDesdoigts
closed
1 year ago
0
Add 3.11 to github workflow tests
#6
LouisDesdoigts
closed
1 year ago
0
`ValueError` when not passing lists into `get_optimiser`
#5
LouisDesdoigts
closed
1 year ago
1
Add `get_jit_model`.
#4
LouisDesdoigts
closed
1 year ago
1
Improve `jit` perfomance using `jax.tree_map`
#3
LouisDesdoigts
closed
1 year ago
1
Setting `None` requires wrapping in a list
#2
LouisDesdoigts
closed
1 year ago
0
doc suggestions
#1
benjaminpope
closed
1 year ago
1