v1.8.0: Merge pull request #267 from google/version_1.8

Version 1.8

v1.7.0: Merge pull request #261 from cyugao/master

Fix jax index update in the tutorial

v1.6.0

Fix a few issue in the code.
Add group norm.
Make DP-SGD privacy accountant to be in sync with Tensorflow Privacy.

v1.4.0: Merge pull request #214 from AlexeyKurakin/release

• Added prototype of ducktyping of Objax variables as JAX arrays
• Added prototype of automatic variable tracing
• Added learning rate scheduler
• Various bugfixes

v1.3.1

• Bugfix: removed accidentally introduced TensorFlow dependency.

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.ExponentialMovingAverageModule for 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

v1.2.0

• Feature: Improved error messages.

• Feature: Extended syntax: allow assigning TrainVar without TrainRef for direction experimentation.

• Feature: Extended padding options or pad and convolution.

• Feature: Modified ResNet_V2 to be Keras compatible.

• Feature: Defaults can be overridden in call for Adam, Momentum.

• BugFix: Layer norm initialization in GPT-2.

v1.1.0: Merge pull request #87 from google/dberth_release_1.1.0

Release branch for 1.1.0

v1.0.2

Updated version number, README.md and index.rst