nnaisense / evotorch
Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.
â1,035Updated 2 months ago
Alternatives and similar repositories for evotorch:
Users that are interested in evotorch are comparing it to the libraries listed below
- â866Updated 7 months ago
- Evolution Strategies in JAX đĻâ539Updated 3 weeks ago
- Hopular: Modern Hopfield Networks for Tabular Dataâ309Updated 2 years ago
- â760Updated last week
- â1,280Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/â1,307Updated this week
- â812Updated last week
- JAX - A curated list of resources https://github.com/google/jaxâ1,676Updated 6 months ago
- RL Environments in JAX đâ682Updated 6 months ago
- Optax is a gradient processing and optimization library for JAX.â1,775Updated this week
- functorch is JAX-like composable function transforms for PyTorch.â1,405Updated this week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â799Updated 5 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ680Updated last month
- TensorDict is a pytorch dedicated tensor container.â868Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.â943Updated 4 months ago
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.â561Updated 3 weeks ago
- Compositional Linear Algebraâ456Updated this week
- A Graph Neural Network Library in Jaxâ1,399Updated 10 months ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/â2,212Updated this week
- â545Updated last month
- A High Level API for Deep Learning in JAXâ472Updated 2 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.â1,415Updated 5 months ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networksâ1,516Updated 4 months ago
- A library to inspect and extract intermediate layers of PyTorch models.â470Updated 2 years ago
- Automatic gradient descentâ206Updated last year
- Accelerated Quality-Diversityâ274Updated 3 months ago
- Really Fast End-to-End Jax RL Implementationsâ800Updated 4 months ago
- A library for distributed ML training with PyTorchâ366Updated 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âĻâ680Updated last year
- JAX-based neural network libraryâ2,945Updated last month