Pinned Loading

1. google/paxml google/paxml Public

   Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…

   Python 538 71

2. pytorch/xla pytorch/xla Public

   Enabling PyTorch on XLA Devices (e.g. Google TPU)

   C++ 2.7k 559

3. google/praxis google/praxis Public

   Python 189 44