elegy 0.8.6


Elegy is a Neural Networks framework based on Jax and Haiku.

Elegy is a Neural Networks framework based on Jax and Haiku.

Stars: 408, Watchers: 408, Forks: 26, Open Issues: 40

The poets-ai/elegy repo was created 2 years ago and was last updated 2 days ago.
The project is popular with 408 github stars!

How to Install elegy

You can install elegy using pip

pip install elegy

or add it to a project with poetry

poetry add elegy

Package Details

Cristian Garcia
GitHub Repo


No  elegy  pypi packages just yet.


A list of common elegy errors.

Code Examples

Here are some elegy code examples and snippets.

GitHub Issues

The elegy package has 40 open issues on GitHub

  • Cleanup Documentation
  • Random state within a Loss
  • [Bug] Installing elegy is broken by jax v0.2.21

See more issues on GitHub

Related Packages & Articles

dm-haiku 0.0.7

Haiku is a library for building neural networks in JAX.

thinc 8.1.1

A refreshing functional take on deep learning, compatible with your favorite libraries

ktrain 0.31.8

ktrain is a wrapper for TensorFlow Keras that makes deep learning and AI more accessible and easier to apply

yapic 1.3.2

Yet another Pixel Classifier (based on deep learning)