awesome-jax
elegy
| awesome-jax | elegy | |
|---|---|---|
| 3 | 5 | |
| 1,312 | 463 | |
| - | 0.0% | |
| 6.2 | 0.0 | |
| 8 days ago | over 1 year ago | |
| Python | ||
| Creative Commons Zero v1.0 Universal | MIT License |
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.
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
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
What are some alternatives?
awesome-production-machine-learning - A curated list of awesome open source libraries to deploy, monitor, version and scale your machine learning
dm-haiku - JAX-based neural network library
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.
jax-resnet - Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
awesome-ocr
equinox - Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
awesome-ai-in-finance - 🔬 A curated list of awesome LLMs & deep learning strategies & tools in financial market.
flax - Flax is a neural network library for JAX that is designed for flexibility.
scenic - Scenic: A Jax Library for Computer Vision Research and Beyond
awesome-deep-learning - A curated list of awesome Deep Learning tutorials, projects and communities.
runtime - A performant and modular runtime for TensorFlow