nnaisense / evotorch
Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.
â1,054Updated 2 weeks ago
Alternatives and similar repositories for evotorch:
Users that are interested in evotorch are comparing it to the libraries listed below
- â897Updated 9 months ago
- Evolution Strategies in JAX đĻâ602Updated 3 weeks ago
- â776Updated 2 weeks ago
- Optax is a gradient processing and optimization library for JAX.â1,868Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.â960Updated last week
- A Graph Neural Network Library in Jaxâ1,425Updated last year
- â845Updated 3 weeks ago
- â1,309Updated last week
- â565Updated last week
- JAX-based neural network libraryâ3,013Updated this week
- JAX - A curated list of resources https://github.com/google/jaxâ1,788Updated 2 months ago
- functorch is JAX-like composable function transforms for PyTorch.â1,422Updated this week
- A High Level API for Deep Learning in JAXâ475Updated 2 years ago
- ML Collections is a library of Python Collections designed for ML use cases.â945Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/â2,324Updated this week
- TensorDict is a pytorch dedicated tensor container.â911Updated this week
- RL Environments in JAX đâ742Updated 9 months ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/â1,375Updated 3 weeks ago
- A platform for managing machine learning experimentsâ846Updated this week
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.â580Updated 2 weeks ago
- a Lightweight library for sequential learning agents, including reinforcement learningâ429Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.â472Updated 2 years ago
- Hopular: Modern Hopfield Networks for Tabular Dataâ311Updated 2 years ago
- The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series oâĻâ721Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ719Updated last month
- Memory mapped numpy arrays of varying shapesâ297Updated 10 months ago
- Turn SymPy expressions into trainable JAX expressions.â334Updated 3 months ago
- PIX is an image processing library in JAX, for JAX.â414Updated last month
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.â1,428Updated 8 months ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networksâ475Updated 2 years ago