|26 days ago||9 days ago|
|MIT License||Apache License 2.0|
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.
[D] What JAX NN library to use?
6 projects | reddit.com/r/MachineLearning | 14 Apr 2022
Equinox! I am of course hugely biased, I am the author of Equinox...6 projects | reddit.com/r/MachineLearning | 14 Apr 2022
See CONTRIBUTING.md for details on getting started, and how to build the documentation locally.
[D] What useful personal projects have you made?
2 projects | reddit.com/r/MachineLearning | 9 Apr 2022
A neural network library for JAX: Equinox. IMO the simplest/most-generic one out there.
[D] Should We Be Using JAX in 2022?
8 projects | reddit.com/r/MachineLearning | 15 Feb 2022
Author of Equinox here. I'm glad to see it being mentioned in the wild!
[D] Current State of JAX vs Pytorch?
3 projects | reddit.com/r/MachineLearning | 1 Feb 2022
There's a good chance that you're interested in building neural networks. In this case be aware that JAX is roughly equivalent to just the torch namespace, and that you can/should choose from various external libraries for building neural networks. The two most popular are Flax and Haiku. Personally I use Equinox which is designed to be a lot more powerful, easier to use, more general etc. (Disclaimer: I am the author of Equinox -- it's something I wrote when I found that Flax/Haiku simply weren't suitable for my use cases.)
[D] Ideal deep learning library
9 projects | reddit.com/r/MachineLearning | 5 Jan 2022
On the assumption that you're doing something neural network related: have a look at the examples section for one of its deep learning libraries. (e.g. this example trains an RNN on a toy classification problem)9 projects | reddit.com/r/MachineLearning | 5 Jan 2022
So I made Equinox! It has a deliberately PyTorch-like feel, keeps init/call separate, and (unlike Haiku etc.) works with native JAX transformations like jit etc. out-of-the-box.
[D] Are you using PyTorch or TensorFlow going into 2022?
6 projects | reddit.com/r/MachineLearning | 14 Dec 2021
I just started using equinox for neural networks in jax. It's really simple. I'd highly recommend you take a look!
[D] Has anyone tried Deepmind haiku?
1 project | reddit.com/r/MachineLearning | 18 Nov 2021
For a JAX neural network library, I'd recommend trying Equinox.
[P] Treex: A Pytree-based Module system for Deep Learning in JAX
2 projects | reddit.com/r/MachineLearning | 5 Sep 2021
Equinox author here. Just chiming in on this comment:
What are some alternatives?
treex - A Pytree Module system for Deep Learning in JAX
flax - Flax is a neural network library for JAX that is designed for flexibility.
fsharp - Native support for destructuring F# types when logging to Serilog.
TF_JAX_tutorials - All about the fundamental blocks of TF and JAX!
Roslyn - The Roslyn .NET compiler provides C# and Visual Basic languages with rich code analysis APIs.
Giraffe - A native functional ASP.NET Core web framework for F# developers.
the-ray-tracer-challenge-fsharp - F# implementations of the ray tracer found in The Ray Tracer Challenge book by Jamis Buck.
torchtyping - Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.