Elegy Alternatives

Similar projects and alternatives to elegy

jax

82 27,936 10.0 Python elegy VS jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
equinox

31 1,809 9.2 Python elegy VS equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
diffrax

21 1,230 8.3 Python elegy VS diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
dm-haiku

10 2,806 7.8 Python elegy VS dm-haiku

JAX-based neural network library
flax

10 5,497 9.6 Python elegy VS flax

Flax is a neural network library for JAX that is designed for flexibility.
jax-resnet

1 98 0.0 Python elegy VS jax-resnet

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
runtime

2 746 9.6 C++ elegy VS runtime

A performant and modular runtime for TensorFlow (by tensorflow)
WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
jax-models

6 138 0.0 Python elegy VS jax-models

Unofficial JAX implementations of deep learning research papers
scenic

5 2,995 8.6 Python elegy VS scenic

Scenic: A Jax Library for Computer Vision Research and Beyond (by google-research)
awesome-jax

3 1,274 6.4 elegy VS awesome-jax

JAX - A curated list of resources https://github.com/google/jax
extending-jax

2 352 3.5 Python elegy VS extending-jax

Extending JAX with custom C++ and CUDA code
long-range-arena

6 682 0.0 Python elegy VS long-range-arena

Long Range Arena for Benchmarking Efficient Transformers

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a better elegy alternative or higher similarity.

Suggest an alternative to elegy

elegy reviews and mentions

Posts with mentions or reviews of elegy. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2022-09-29.

is Elegy framework for JAX abandoned?
1 project | /r/deeplearning | 21 Feb 2023

I wonder if https://github.com/poets-ai/elegy is still an active project or dead because it hasn't had a commit in almost a year. Would be too bad if abandoned because I like it.
[D] Any less-boilerplate framework for Jax/Flax/Haiku?
4 projects | /r/MachineLearning | 29 Sep 2022

Elegy might be worth a look.
PyTorch vs. TensorFlow in Academic Papers
2 projects | news.ycombinator.com | 13 Mar 2022

JAX is really cool, but still somewhat immature. I would love to see it taking more ground and improving wrt e.g. integration with tensorboard and getting all the goodies we have in tensorflow. If you are looking for a higher level framework, I would recommend elegy [0] which is very close to the keras API.
[0] https://github.com/poets-ai/elegy
[D] Should We Be Using JAX in 2022?
8 projects | /r/MachineLearning | 15 Feb 2022

What's your favorite Deep Learning API for JAX - Flax, Haiku, Elegy, something else?
Best sources to learn JAX?
1 project | /r/JAX | 2 Mar 2021

For a Module library checkout Flax or Haiku, they are well maintained. For a Trainer interface like Keras / Pytorch Lightning checkout Elegy: https://github.com/poets-ai/elegy
A note from our sponsor - WorkOS
workos.com | 25 Apr 2024

The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning. Learn more →