search

google / objax

Apache License 2.0
769 stars 77 forks source link

Remove BaseState limitations in JAX wrapping. #121

Closed david-berthelot closed 3 years ago

david-berthelot commented 3 years ago

Resolves #114