flax

db5b63c0 - Refactor jit explanation, use parameters instead of weights in text