Deep memory and sequence models in JAX
☆25Jan 15, 2026Updated 2 months ago
Alternatives and similar repositories for memax
Users that are interested in memax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- Official implementation for "How Should We Meta-Learn Reinforcement Learning Algorithms?"☆23Sep 7, 2025Updated 7 months ago
- Evidential Calibration☆11Mar 8, 2022Updated 4 years ago
- Reward Evolution with Large Language Models using Human Feedback☆18Nov 14, 2025Updated 4 months ago
- Official implementation of DiscoGen, for "Procedural Generation of Algorithm Discovery Tasks in Machine Learning"☆31Mar 27, 2026Updated 2 weeks ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A neural network library written in jax☆13Feb 3, 2025Updated last year
- DifferentialEquations.jl with PyTorch☆11Oct 12, 2022Updated 3 years ago
- Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…☆11Dec 19, 2025Updated 3 months ago
- A codebase dedicated to exploring multimodal learning approaches by integrating images of host galaxies of supernovae and their correspon…☆17Jan 15, 2025Updated last year
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆403Mar 18, 2026Updated 3 weeks ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated 10 months ago
- Accelerated minigrid environments with JAX☆164Oct 20, 2025Updated 5 months ago
- Object-oriented Jax framework extending Equinox for scientific programming☆25Mar 20, 2026Updated 3 weeks ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆23Jun 11, 2025Updated 10 months ago
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- ☆29Nov 5, 2025Updated 5 months ago
- A library for language transfer methods and algorithms.☆16Feb 6, 2026Updated 2 months ago
- Materials for a 'Python for Science' bootcamp workshop.☆13Sep 23, 2018Updated 7 years ago
- A tool for aggregating and plotting MARL experiment data.☆84Jan 26, 2026Updated 2 months ago
- suffix array construction and searching algorithms for in-memory binary data.☆12Sep 10, 2022Updated 3 years ago
- Codes for the paper: Huang C, Ruff DA, Pyle R, Rosenbaum R, Cohen MR and Doiron B (2019) “Circuit models of low dimensional shared varia…☆11May 18, 2020Updated 5 years ago
- Rubin Data Academy, a virtual workshop for working with Rubin data products☆36Mar 30, 2026Updated last week
- Python module to remove wiki markup text.☆10Jan 15, 2016Updated 10 years ago
- Demo on chaining together LFP simulation (neurodsp), spectral parameterization (fooof), and simulation-based inference (sbi).☆13Jul 8, 2025Updated 9 months ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Load motion capture data into Roboschool/MuJoCo☆17Oct 28, 2017Updated 8 years ago
- A toolbox for brain network modeling on GPUs☆20Mar 25, 2026Updated 2 weeks ago
- A LaTeX document class for notes 📝 and textbooks 📚☆14Jul 14, 2021Updated 4 years ago
- ReDo: The Dormant Neuron Phenomenon in Deep Reinforcement Learning (pytorch)☆33Oct 22, 2024Updated last year
- A numpyro implementation of BayeSN, the optical-NIR SED model for type Ia supernovae☆17Mar 31, 2026Updated last week
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆22Nov 29, 2025Updated 4 months ago
- ☆41Oct 23, 2025Updated 5 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆103Jan 23, 2025Updated last year
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆38Jun 3, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Master thesis: Exploring bias in German NLG (GPT-3 & GerPT-2). Applies regard classification and bias mitigation triggers.☆16Sep 25, 2024Updated last year
- Benchmarks for the FluxML ecosystem for deep learning, scientific machine learning, differentiable programming etc including AD and CUDA …☆15Jun 4, 2022Updated 3 years ago
- The assignment of a transportation course for my first year as a graduate student, solved the bilevel programming problem using GA algori…☆24Mar 18, 2022Updated 4 years ago
- ☆13Mar 8, 2023Updated 3 years ago
- ☆95Jan 21, 2026Updated 2 months ago
- ✂️ Sentence segmentation with wtpsplit's state-of-the-art Segment any Text (SaT) models☆38Oct 1, 2025Updated 6 months ago
- Official repository for "Temporal Disentanglement of Representations for Improved Generalisation in Reinforcement Learning".☆13Jan 25, 2023Updated 3 years ago