elegy
awesome-jax
elegy | awesome-jax | |
---|---|---|
5 | 3 | |
463 | 1,288 | |
0.0% | - | |
0.0 | 6.2 | |
over 1 year ago | 13 days ago | |
Python | ||
MIT License | Creative Commons Zero v1.0 Universal |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
elegy
-
is Elegy framework for JAX abandoned?
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?
Elegy might be worth a look.
-
PyTorch vs. TensorFlow in Academic Papers
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?
What's your favorite Deep Learning API for JAX - Flax, Haiku, Elegy, something else?
-
Best sources to learn JAX?
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
awesome-jax
-
[D] Any less-boilerplate framework for Jax/Flax/Haiku?
Have you looked in here?
-
[D] JAX learning resources?
Here's a compilation of resources: https://github.com/n2cholas/awesome-jax
What are some alternatives?
dm-haiku - JAX-based neural network library
awesome-production-machine-learning - A curated list of awesome open source libraries to deploy, monitor, version and scale your machine learning
jax-resnet - Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
get-started-with-JAX - The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
equinox - Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
awesome-ocr
flax - Flax is a neural network library for JAX that is designed for flexibility.
awesome-ai-in-finance - 🔬 A curated list of awesome LLMs & deep learning strategies & tools in financial market.
scenic - Scenic: A Jax Library for Computer Vision Research and Beyond
runtime - A performant and modular runtime for TensorFlow
awesome-deep-learning - A curated list of awesome Deep Learning tutorials, projects and communities.