Alternatives and similar repositories for mistral-jax:

Users that are interested in mistral-jax are comparing it to the libraries listed below

• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• danijar / ninjax
  General Modules for JAX
  ☆64Updated 2 weeks ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• danijar / elements
  Building blocks for productive research
  ☆52Updated 2 months ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆56Updated 2 years ago
• google-deepmind / nao_top10
  ☆18Updated 2 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 5 months ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated 9 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 5 months ago
• keraJLi / synthetic-gymnax
  ☆20Updated 10 months ago
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 2 years ago
• astanic / crafter-ood
  ☆20Updated 2 years ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆14Updated 3 weeks ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆9Updated last year
• ucl-dark / pax
  Scalable Opponent Shaping Experiments in JAX
  ☆24Updated last year
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆53Updated last year
• DramaCow / jaxued
  ☆76Updated last month
• google-deepmind / enn_acme
  ☆31Updated 2 years ago
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆21Updated 2 years ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆27Updated 5 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆28Updated 10 months ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆48Updated last year
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆28Updated 8 months ago
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆22Updated 4 months ago
• haoliuhl / tux
  Tools and Utils for Experiments (TUX)
  ☆16Updated 2 months ago
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago