https://github.com/birkhoffg/haiku-trainer

https://github.com/birkhoffg/haiku-trainer

Last synced: 9 months ago
JSON representation

• Host: GitHub
• URL: https://github.com/birkhoffg/haiku-trainer
• Owner: BirkhoffG
• License: apache-2.0
• Archived: true
• Created: 2023-06-26T07:01:07.000Z (almost 3 years ago)
• Default Branch: main
• Last Pushed: 2023-07-07T20:56:03.000Z (almost 3 years ago)
• Last Synced: 2025-04-07T06:22:11.332Z (about 1 year ago)
• Language: Python
• Size: 768 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# haiku-trainer

![CI](https://github.com/BirkhoffG/haiku-trainer/actions/workflows/test.yaml/badge.svg)

![Docs](https://github.com/BirkhoffG/haiku-trainer/actions/workflows/deploy.yaml/badge.svg)

![License](https://img.shields.io/github/license/BirkhoffG/haiku-trainer.svg)

`haiku-trainer` is an experimental package to facilitate fast,

extensible, and scalable deep learning in `dm-haiku` and `JAX`.

## Install

``` sh

pip install haiku_trainer

```