⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
MIT License
Accurate sums and dot products for Python.
np.einsum but better
Optax is a gradient processing and optimization library for JAX.
Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021
Use scipy.optimize.minimize as a PyTorch Optimizer.
Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
A hyperparameter optimization framework
Hyper optimized contraction trees for large tensor networks and einsums
End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 ...
CLI & Python API to easily summarize text-based files with transformers
One tensor operation is all you need
🐦 Opytimizer is a Python library consisting of meta-heuristic optimization algorithms.