Alternatives and similar repositories for JAXAtari

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

• sdevries0 / Kozax
  A JAX-based framework for genetic programming
  ☆24Updated last month
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆359Updated this week
• patrick-kidger / diffrax
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,785Updated last week
• SymposiumOrganization / ControllableNeuralSymbolicRegression
  ☆12Updated 2 years ago
• danielward27 / flowjax
  ☆189Updated 2 months ago
• facebookresearch / symbolicregression
  SR based on LLMs.
  ☆114Updated 2 years ago
• google-deepmind / distrax
  ☆597Updated this week
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆290Updated 2 weeks ago
• GilpinLab / dysts
  Hundreds of strange attractors
  ☆495Updated 2 weeks ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆813Updated 4 months ago
• FAIR-Universe / HEP-Challenge
  This is the repository for Fair Universe Uncertainty Challenge
  ☆14Updated 4 months ago
• josephbakarji / deep-delay-autoencoder
  Discovers high dimensional models from 1D data using deep delay autoencoders
  ☆37Updated 2 years ago
• RobertTLange / evosax
  Evolution Strategies in JAX 🦎
  ☆668Updated 3 weeks ago
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,541Updated last week
• SymposiumOrganization / NeuralSymbolicRegressionThatScales
  Source code and Dataset creation for the paper "Neural Symbolic Regression That Scales"
  ☆100Updated last month
• SJ001 / AI-Feynman
  ☆711Updated last year
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆253Updated 3 weeks ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆540Updated this week
• dso-org / deep-symbolic-optimization
  A deep learning framework for symbolic optimization.
  ☆681Updated 9 months ago
• patrick-kidger / torchcde
  Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
  ☆452Updated last month
• raminmh / liquid-s4
  Liquid Structural State-Space Models
  ☆371Updated last year
• lindermanlab / S5
  ☆303Updated 9 months ago
• probml / dynamax
  A Python package for probabilistic state space modeling with JAX
  ☆871Updated 3 months ago
• CSML-IIT-UCL / linear_operator_learning
  A package to Learn Linear Operators
  ☆13Updated 4 months ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆518Updated 5 months ago
• absorensen / numbers_go_brrrr
  They do indeed go brrrr
  ☆30Updated 4 months ago
• automl / TransformersCanDoBayesianInference
  Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper
  ☆233Updated 11 months ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆117Updated 5 months ago
• TyXe-BDL / TyXe
  ☆154Updated 3 years ago
• LaurenceA / infrastructure
  ☆15Updated last month