jax-models
extending-jax
| jax-models | extending-jax | |
|---|---|---|
| 6 | 2 | |
| 138 | 353 | |
| - | - | |
| 0.0 | 3.5 | |
| almost 2 years ago | 7 months ago | |
| Python | Python | |
| Apache License 2.0 | 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.
jax-models
-
[D] How to contribute to open source ML and DL without having access to high quality setup?
I was in the same position as you are and the best thing you can do is to start reproducing papers (that's what I did with jax-models). This will
-
[D] Should We Be Using JAX in 2022?
I've been using JAX, especially Flax for quite some time now for my reproducibility initiative (jax_models) and this is what I really appreciate about the framework
- Weekly updated open sourced model implementations in Flax
- Weekly updated open sourced deep learning model implementations in Flax
- [P] Weekly updated open sourced model implementations in Flax
extending-jax
-
[D] Should We Be Using JAX in 2022?
You can check out this or this for more info. I think it is safe to assume that it is less stable than PyTorch - some other commenters have spoken about running into trouble with XLA in certain corner cases, but I have not experienced this so I can't speak to it.
- Extending JAX with custom C++ and CUDA code
What are some alternatives?
datasets - TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
einops - Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
flax - Flax is a neural network library for JAX that is designed for flexibility.
mpi4jax - Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
flaxmodels - Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
thinc - 🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
equinox - Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
GradCache - Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
trax - Trax — Deep Learning with Clear Code and Speed
elegy - A High Level API for Deep Learning in JAX