Skip to content

Commit

Permalink
Internal change.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 454961096
  • Loading branch information
pax authors authored and laurentes committed Jun 14, 2022
1 parent 9cdf644 commit e27d4be
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ What is Praxis?
Praxis is the layer library for [Pax](https://github.com/google/paxml/). While Praxis is optimized for ML at scale, Praxis has a goal to be usable by other JAX-based ML projects.


Some examples of layers to be folded into Praxis are in the [praxis/layers/](https://github.com/google/praxis/praxis/layers/) directory.
Some examples of layers to be folded into Praxis are in the [praxis/layers/](https://github.com/google/praxis/tree/main/praxis/layers) directory.


Copyright 2022 Google LLC
Expand Down
4 changes: 2 additions & 2 deletions praxis/setup.py → praxis/pip_package/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,9 @@
packages=find_packages(),
python_requires='>=3.7',
install_requires=[
'protobuf', 'absl-py', 'fiddle', 'numpy', 'tensorflow', 'lingvo',
'protobuf', 'absl-py', 'fiddle-config', 'numpy', 'tensorflow', 'lingvo',
'flax', 'jax', 'optax', 'optax-shampoo', 'jax-bitempered-loss',
'einops', 'flaxformer', 'fiddle', 't5x', 'clu'
'einops', 't5x', 'clu'
],
url='https://github.com/google/praxis',
license='Apache-2.0',
Expand Down

0 comments on commit e27d4be

Please sign in to comment.