torchopt
jaxopt
| torchopt | jaxopt | |
|---|---|---|
| 1 | 1 | |
| 501 | 891 | |
| 4.4% | 1.2% | |
| 7.1 | 7.8 | |
| 2 days ago | 17 days ago | |
| Python | Python | |
| Apache License 2.0 | 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.
torchopt
-
What is the most efficient approach to ensemble a pytorch actor-critic model?
I would suggest checking https://pytorch.org/functorch/ and https://github.com/metaopt/torchopt for efficient inference and training with ensembles (e.g., t be independent actors in a multi-agent setting or multiple critics).
jaxopt
What are some alternatives?
Adan-pytorch - Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
symbolicai - Compositional Differentiable Programming Library
einops - Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
theseus - A library for differentiable nonlinear optimization
datasets - TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
functorch - functorch is JAX-like composable function transforms for PyTorch.
pyprobml - Python code for "Probabilistic Machine learning" book by Kevin Murphy
pennylane - PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.
PyNeuraLogic - PyNeuraLogic lets you use Python to create Differentiable Logic Programs