Alternatives and similar repositories for JAXAtari

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

• RobertTLange / evosax
  Evolution Strategies in JAX 🦎
  ☆679Updated last month
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆821Updated 5 months ago
• patrick-kidger / diffrax
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,810Updated last month
• sdevries0 / Kozax
  A JAX-based framework for genetic programming
  ☆26Updated 3 weeks ago
• pdearena / pdearena
  ☆273Updated last year
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆365Updated this week
• wu375 / simple-physics-simulator-pytorch-geometry
  Minimal pytorch version of https://github.com/deepmind/deepmind-research/tree/master/learning_to_simulate
  ☆52Updated 4 years ago
• Rubiksman78 / PINN_torch
  Pytorch implementation of Physics Informed Neural Networks and improvements
  ☆14Updated 2 years ago
• MatthieuDarcy / KernelsOperatorLearning
  ☆11Updated 10 months ago
• PredictiveIntelligenceLab / jaxpi
  ☆368Updated last month
• camlab-ethz / ConvolutionalNeuralOperator
  This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs
  ☆195Updated 5 months ago
• spdes / computational-sde-intro-lecture
  Lecture "A computational introduction to stochastic differential equations".
  ☆27Updated 4 months ago
• ScottAlexanderCameron / Jynx
  A neural network library written in jax
  ☆12Updated 9 months ago
• GilpinLab / dysts
  Hundreds of strange attractors
  ☆498Updated last month
• pdebench / PDEBench
  PDEBench: An Extensive Benchmark for Scientific Machine Learning
  ☆986Updated 5 months ago
• limix / chi2comb-py
  Python package for linear combination of independent noncentral chi-squared random variables.
  ☆11Updated 5 years ago
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆256Updated last month
• zabaras / transformer-physx
  Transformers for modeling physical systems
  ☆145Updated 2 years ago
• instadeepai / jumanji
  🕹️ A diverse suite of scalable reinforcement learning environments in JAX
  ☆769Updated this week
• brandstetter-johannes / MP-Neural-PDE-Solvers
  Repo to the paper "Message Passing Neural PDE Solvers"
  ☆139Updated last year
• excitingtimes / awesome-survey-papers
  A collection of Survey Papers in Machine Learning
  ☆20Updated 3 years ago
• google-deepmind / distrax
  ☆602Updated this week
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,637Updated last week
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆293Updated last week
• tum-pbs / ConFIG
  [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method
  ☆84Updated 5 months ago
• awwang10 / sphinx
  ☆12Updated last week
• raminmh / liquid-s4
  Liquid Structural State-Space Models
  ☆374Updated last year
• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆966Updated last year
• bwfbowen / muax
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆63Updated 11 months ago
• karush17 / Deep-Eligibility-Traces
  Implementation of Eligibility Traces with Neural Networks in PyTorch and Tensorflow 2.0
  ☆26Updated 4 years ago