Alternatives and similar repositories for torchopt:

Users that are interested in torchopt are comparing it to the libraries listed below

• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆876Updated this week
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆946Updated 4 months ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,408Updated this week
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆572Updated last month
• facebookresearch / amortized-optimization-tutorial
  Tutorial on amortized optimization for learning to optimize over continuous domains
  ☆236Updated last year
• google-deepmind / distrax
  ☆546Updated 2 months ago
• RobertTLange / evosax
  Evolution Strategies in JAX 🦎
  ☆546Updated last week
• google-research / rliable
  [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.
  ☆800Updated 6 months ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆165Updated this week
• google-deepmind / chex
  ☆822Updated this week
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆303Updated 2 months ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆666Updated 2 years ago
• ikostrikov / jaxrl
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆654Updated 2 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆566Updated this week
• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆808Updated 5 months ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆366Updated 2 years ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆360Updated this week
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆248Updated 4 months ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆696Updated 7 months ago
• DiffEqML / torchdyn
  A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
  ☆1,438Updated 9 months ago
• facebookresearch / mbrl-lib
  Library for Model Based RL
  ☆983Updated 7 months ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆460Updated last week
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,798Updated this week
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆235Updated last year
• sail-sg / envpool
  C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.
  ☆1,121Updated 6 months ago
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,316Updated this week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆219Updated this week
• nnaisense / evotorch
  Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.
  ☆1,038Updated last week
• instadeepai / jumanji
  🕹️ A diverse suite of scalable reinforcement learning environments in JAX
  ☆685Updated last week