Metadata-Version: 2.1
Name: bobbin
Version: 0.0.2
Summary: Tools for making training loops with flax.linen models.
Home-page: https://github.com/yotarok/bobbin
Author: Bobbin Team
Author-email: yotaro@google.com
Keywords: python machine learning flax
Classifier: Development Status :: 1 - Planning
Classifier: Environment :: Console
Classifier: Intended Audience :: Science/Research
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.8
Description-Content-Type: text/markdown
License-File: LICENSE

# Bobbin

![Unittest](https://github.com/yotarok/bobbin/actions/workflows/lint_and_test.yml/badge.svg)
[![Documentation Status](https://readthedocs.org/projects/bobbin/badge/?version=latest)](http://bobbin.readthedocs.io)

## What is this?

A small tool for making neural net training loop with flax.linen.

This library is designed to be a set of small tools and not to be a full-blown
framework for deep learning.  Users should keep freedom not to use some parts
of this library and the rest of the library should still be able to serve the
users.

Currently, this library contains the following components:

- `cron`:
  Tools for defining and performing periodical actions in the training-loop.
- `evaluation`:
  Helpers for performing evaluation and handling metrics.
- `pmap_utils`:
  Tools for helping the use of `jax.pmap`.
- `pytypes`:
  Type definitions for better type annotations.
- `tensorboard`:
  Helper functions for publishing training state and evaluation summaries.
- `training`:
  Helpers for writing training-loops for Flax models.
- `var_utils`:
  Utility for accessing variable collections of Flax models.


## Contributing

See [`CONTRIBUTING.md`](CONTRIBUTING.md) for details.

## License

Apache 2.0; see [`LICENSE`](LICENSE) for details.

## Disclaimer

This library is still in a very early phase of development, and therefore,
its API can often be subject to destructive change.

This project is not an official Google project. It is not supported by
Google and Google specifically disclaims all warranties as to its quality,
merchantability, or fitness for a particular purpose.
