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

The poets-ai/elegy repo was created 3 years ago and the last code push was 1 years ago.
The project is popular with 468 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

thinc 9.0.0

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

keras 3.3.3

Keras is a deep learning API written in Python, running on top of the machine learning platform TensorFlow. The core data structures of Keras are layers and models. The philosophy is to keep simple things simple, while allowing the user to be fully in control when they need to (the ultimate control being the easy extensibility of the source code via subclassing).

ktrain 0.41.3

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