This repository has been archived by the owner on Jul 14, 2024. It is now read-only.

refactor ekf_mlp to work with flax models #34

Closed

murphyk opened this issue Apr 13, 2022 · 1 comment

Assignees

Member

murphyk commented Apr 13, 2022

After solving #33, modify ekf_mlp so it also works with any flax regression model

murphyk assigned karalleyna

Collaborator

karalleyna commented Apr 15, 2022

JSL/jsl/demos/ekf_mlp.py

Line 38 in 3e13b32

def apply(flat_params, x, model, unflatten_fn):

karalleyna closed this as completed

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.