Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)
☆16Aug 6, 2024Updated 2 years ago
Alternatives and similar repositories for xLSTM-Jax
Users that are interested in xLSTM-Jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆16May 28, 2025Updated last year
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago
- This is a port of Mistral-7B model in JAX☆34Jul 1, 2024Updated 2 years ago
- Disciplined Convex Programming in R using Convex.jl.☆14Dec 18, 2018Updated 7 years ago
- Plotutils+ is a fork of the GNU plotutils package, with fixes and extensions☆13May 14, 2024Updated 2 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- A set of jupyter notebooks☆27Dec 18, 2024Updated last year
- Differentiable interface to Firedrake for JAX☆15Feb 28, 2021Updated 5 years ago
- Simple process engineering calculation with Python☆23Jul 30, 2020Updated 6 years ago
- Gymnasium environment for research of UAVs and risk constraints☆13Oct 29, 2024Updated last year
- Full Bayesian inference for neural networks using TensorFlow☆15Aug 19, 2021Updated 5 years ago
- Python Implementation of Lattice Boltzmann Method☆12Aug 18, 2021Updated 5 years ago
- Physics-guided Deep Markov Models☆13May 24, 2022Updated 4 years ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- Code for "What really matters in matrix-whitening optimizers?"☆25Oct 31, 2025Updated 11 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago
- Implementation of Flow Matching model for MNIST to understand how it works☆17Sep 19, 2025Updated last year
- Code for "Deep Energy-Based Modeling of Discrete-Time Physics," NeurIPS, 2020. (Oral)☆19Jan 30, 2022Updated 4 years ago
- ☆10Jun 13, 2023Updated 3 years ago
- OpenAI GYM environment for 6-DOF Helicopter simulation☆20Apr 27, 2024Updated 2 years ago
- Practical Deep Reinforcement Learning with Python, by BPB Publications☆23Jul 21, 2022Updated 4 years ago
- ⚡️PyTorch Lightning Implementations of Recent Low-Light Image Enhancement !☆17Sep 19, 2022Updated 4 years ago
- Automatic Differentiation for R☆23Dec 18, 2018Updated 7 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- SeismicNet: Physics-informed neural networks for seismic wave modeling in semi-infinite domain☆22May 5, 2024Updated 2 years ago
- R package to pixelate and map spatially continuous point estimates according to the uncertainty that surrounds them☆29Jul 7, 2022Updated 4 years ago
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆15Jun 24, 2023Updated 3 years ago
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆52Apr 26, 2024Updated 2 years ago
- A Survey Analyzing Generalization in Deep Reinforcement Learning☆36Oct 31, 2024Updated last year
- The official code release for "Langevin Soft Actor-Critic: Efficient Exploration through Uncertainty-Driven Critic Learning", ICLR 2025☆23May 28, 2025Updated last year
- ☆10Oct 1, 2019Updated 7 years ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Jul 18, 2023Updated 3 years ago
- This project contains the UNIFIER19 C7A-HARW aircraft Flight Dynamics Simulator Simulink model and XPlane-11 model.☆27Oct 14, 2024Updated 2 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆18Oct 22, 2024Updated last year
- A simple, continuous-control environment for OpenAI Gym☆23Jan 1, 2023Updated 3 years ago
- Official repository for paper "Conservative Offline Distributional Reinforcement Learning" (NeurIPS 2021)☆22Aug 1, 2021Updated 5 years ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- Keras implementation of InfoGAN (work in progress)☆11May 25, 2017Updated 9 years ago
- KernelBench v2: Can LLMs Write GPU Kernels? - Benchmark with Torch -> Triton (and more!) problems☆26Jul 4, 2025Updated last year
- This Python-based simulator models the longitudinal flight dynamics of fixed-wing UAVs using a linear state-space approach. It allows aer…☆26Sep 22, 2025Updated last year