jheek

Fix missing seperator in static fold in

1

fixes #2157

checkpoints.restore_checkpoint` should raise an error for explicit paths that don't exist

2

### Discussed in https://github.com/google/flax/discussions/1612 Originally posted by **PgLoLo** October 9, 2021 From documentation of `flax.training.checkpoints.restore_checkpoint`: >Returns: > Restored `target` updated from checkpoint file, or if no step specified and >...

Example template

1

Currently there's no single source of truth for the patterns that an example should adhere to. With a blank example template we could more easily keep the patterns up-to-date and...

Pytype attribute generics tracker

3

Pytype currently doesn't support Generic types class attributes: ``` class Foo: bar: T ``` As a workaround we use `Any` for the attribute instead. This workaround should be reverted once...

Respecting input dtype

2

Currently we use a default `dtype=float32` in the standard layer implementations. We would like to switch this to something more closely resembling numpy semantics. This means the default is `dtype=None`...

FLIP: General metadata

1

Issue #2434

See PR #2435 [rendered](https://github.com/google/flax/pull/2435/files?short_path=e1c0a0e#diff-e1c0a0e3387b6227be768dc5a1163e254309805048393ec5ce33dde982dca0cc)

Test pickle support

1

PR #1475 fixes cloudpickle support but doesn't support add any testing. Our current test setup seems to import dill indirectly which is incompatible with cloudpickle. Ideally we want to setup...