This repository was archived by the owner on Mar 31, 2025. It is now read-only.
v1.3.0
- Feature: Improved error messages overall
- Feature: Improved BatchNorm numerical stability
- Feature: Objax2Tf for serving objax using TensorFlow
- Feature: New API
objax.optimizer.ExponentialMovingAverageModulefor easy moving average of a model - Feature: Automatic broadcasting of scalars for objax.Parallel
- Feature: New optimizer: LARS
- Feature: New API added to functional (lax.scan)
- Feature: Modules can be printed to nicely readable text now (repr)
- Feature: New interpolate API (for images)
- Bugfix: make objax.Sequential work with latest JAX