jaxopt
torchopt
Our great sponsors
jaxopt | torchopt | |
---|---|---|
1 | 1 | |
888 | 496 | |
1.6% | 5.0% | |
7.8 | 7.1 | |
2 days ago | 26 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.
jaxopt
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).
What are some alternatives?
jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Adan-pytorch - Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
einops - Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
symbolicai - Compositional Differentiable Programming Library
pyprobml - Python code for "Probabilistic Machine learning" book by Kevin Murphy
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.
PyNeuraLogic - PyNeuraLogic lets you use Python to create Differentiable Logic Programs
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.