8000 Commits · google/paxml · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on Apr 7, 2023

pin tfds-nightly
aireenmei
committed

Commits on Apr 6, 2023

pin tensorflow-datasets to 4.8.3
aireenmei
committed

Commits on Mar 31, 2023

Fix condition in Dockerfile

A9isha
authored and
aireenmei
committed
update release test to include Praxis from Github

A9isha
authored and
aireenmei
committed

Commits on Mar 30, 2023

update fiddle version in requirements.txt
aireenmei
committed
praxis/paxml 0.4.0 release

The paxml Authors
authored and
aireenmei
committed

Commits on Mar 29, 2023

Support CLU audio metrics summary in pax
The paxml Authors
committed

Commits on Mar 28, 2023

Update requirements.txt following new jaxlib version.

laurentes
authored and
pax authors
committed
[pax] Remove input_p from _common_eval_or_decode_loop

ppwwyyxx
authored and
pax authors
committed
Move tasks_lib.SingleTask.early_stopping_fn to direct instantiation. #fiddle
The paxml Authors
committed
Use PaxCheckpointHandler for restoring checkpoints when pmap_use_tensorstore is True. This will be useful as we prepare to unify checkpointing via OCDBT.

cpgaffney1
authored and
pax authors
committed
Avoid passing the checkpoint_type around and get it using the checkpointer.

aaroey
authored and
pax authors
committed
update path for requirements.in in docker
The paxml Authors
committed
fix copybara to move setup.py and requirement.in to root
The paxml Authors
committed
move setup.py and requirements.in to root to support installing from github
The paxml Authors
committed
[pax] Remove input_p from decode_once_{pmap,spmd}_model

ppwwyyxx
authored and
pax authors
committed

Commits on Mar 27, 2023

Move AsyncCheckpointer initialization into train.py. There is no real reason for it to happen in main.py anymore - this is left over from pre-Orbax days. It is preferable to consolidate checkpointi…

cpgaffney1
authored and
pax authors
committed

Commits on Mar 25, 2023

Clean-up residual of flaxformer.

laurentes
authored and
pax authors
committed
remove t5x from requirements
The paxml Authors
committed

Commits on Mar 24, 2023

Respect optional job log dir in partitioning.

Haitang Hu
authored and
pax authors
committed
Fix pytype failures related to teaching pytype about NumPy scalar types.

hawkinsp
authored and
pax authors
committed
Merge _PmapEvalRunner and _SpmdEvalRunner in to a single _EvalRunner.

aaroey
authored and
pax authors
committed

Commits on Mar 23, 2023

Refactor _PmapEvalRunner and _SpmdEvalRunner and make sure they have the same API.

aaroey
authored and
pax authors
committed
Implement PmapPartitioner.preprocess_prng_key() and use it in the pmap

aaroey
authored and
pax authors
committed
Clean up `hasattr` usage now that Orbax provides the required function in the open-source package.

cpgaffney1
authored and
pax authors
committed

Commits on Mar 22, 2023

Make release pipeline read updated requirements.in

A9isha
authored and
pax authors
committed
Move learners.Learner.stochastic_gradient to direct instantiation. #fiddle
The paxml Authors
committed
update requirements.txt

A9isha
authored and
pax authors
committed
Cleanup: remove the train state partition spec return value from initialize_partitioned_model_states(), an use partitioner to provide the partition specs.

aaroey
authored and
pax authors
committed
Update requirements.txt

A9isha
authored and
pax authors
committed

Commits on Mar 21, 2023

Introduce a partitioner method to initialize the train states and root prng

aaroey
authored and
pax authors
committed
Call jax.distributed.initialize() more conditionally

laurentes
authored and
pax authors
committed
Remove redundant train_input property definition.

aaroey
authored and
pax authors
committed
Merge pull request #13 from ashors1:gpu_dependencies_and_scripts
pax authors
committed
Merge pull request #8 from ashors1:update_metrics_and_gpu_flags
pax authors
committed

0