https://github.com/google/praxis

https://github.com/google/praxis

Last synced: 3 months ago
JSON representation

• Host: GitHub
• URL: https://github.com/google/praxis
• Owner: google
• License: apache-2.0
• Created: 2022-06-14T16:04:46.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-08-02T03:48:44.000Z (3 months ago)
• Last Synced: 2024-08-02T04:58:45.597Z (3 months ago)
• Language: Python
• Size: 3.85 MB
• Stars: 166
• Watchers: 8
• Forks: 41
• Open Issues: 22
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-jax - Praxis - The layer library for Pax with a goal to be usable by other JAX-based ML projects. <img src="https://img.shields.io/github/stars/google/praxis?style=social" align="center"> (Libraries / New Libraries)

README

        
# Praxis

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/tree/main/praxis/layers) directory.

    Copyright 2022 Google LLC

    Licensed under the Apache License, Version 2.0 (the "License");

    you may not use this file except in compliance with the License.

    You may obtain a copy of the License at

        https://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software

    distributed under the License is distributed on an "AS IS" BASIS,

    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

    See the License for the specific language governing permissions and

    limitations under the License.