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: 435, Watchers: 435, Forks: 32, Open Issues: 46

The poets-ai/elegy repo was created 2 years ago and was last updated 9 hours ago.
The project is popular with 435 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 46 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.9

Haiku is a library for building neural networks in JAX.

thinc 8.1.6

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

ktrain 0.32.3

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)