Skip to content
Snippets Groups Projects
Commit a5ada1f8 authored by Filippo Vicentini's avatar Filippo Vicentini
Browse files

Use jax tree utils if jax is available instead of our fallbacks 

necessary for compatibility with flax, as it uses it's own custom dict
fix to vmc common
parent d3ab5e4d
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
Showing
with 94 additions and 70 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment