Move to flax - Githubissues

Codecov Report

Patch coverage is 68.60% of modified lines.

Files Changed	Coverage
...imental/gaussian_process/train_gaussian_process.py	`22.80%`
...bayesian_neural_network/bayesian_neural_network.py	`38.23%`
...imental/bayesian_neural_network/bayesian_linear.py	`39.39%`
...rimental/gaussian_process/kernel/non_stationary.py	`45.45%`
ramsey/data.py	`47.61%`
ramsey/_src/attention/dotproduct_attention.py	`50.00%`
...experimental/gaussian_process/kernel/stationary.py	`50.00%`
...mental/gaussian_process/sparse_gaussian_process.py	`51.42%`
.../experimental/gaussian_process/gaussian_process.py	`52.38%`
...l/timeseries/recurrent_attentive_neural_process.py	`56.25%`
... and 15 more

:loudspeaker: Thoughts on this report? Let us know!.