ymahlau / albatross
☆22Updated 11 months ago
Alternatives and similar repositories for albatross
Users that are interested in albatross are comparing it to the libraries listed below
Sorting:
- Code for the ICLR 2025 Paper "QPM: Discrete Optimization for Globally Interpretable Image Classification"☆19Updated 3 months ago
- A rich, visual interface for easily starting and monitoring your Hydra applications on SLURM clusters☆14Updated 10 months ago
- A simple to set up 2D automated vehicle gym env featuring random problem generation and dynamic observations.☆13Updated last month
- Code for the AAAI 2024 Paper "Q-SENN: Quantized Self-Explaining Neural Networks"☆21Updated 3 months ago
- Official implementation of "Learned Fourier Bases for Deep Set Feature Extractors in Automotive Reinforcement Learning"☆13Updated last year
- The code for the voraus-AD dataset paper.☆31Updated last year
- Official implementation of Constrained Mean Shift Clustering☆11Updated 3 years ago
- Official implementation of "Deep Reinforcement Learning for Autonomous Driving using High-Level Heterogeneous Graph Representations"☆28Updated last year
- Electromagnetic FDTD Simulations in JAX☆104Updated this week
- Official PyTorch implementation for ChimeraMix: Image Classification on Small Datasets via Masked Feature Mixing (IJCAI 2022)☆14Updated last year
- A collection of matrix games in JAX☆11Updated 5 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆10Updated last year
- PyTorch implementation of the VIME paper (Variational Information Maximizing Exploration)☆7Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆52Updated last year
- Platform for training generalizable deep reinforcement learning agents☆9Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 6 months ago
- ☆17Updated last year
- ☆41Updated 3 years ago
- ☆77Updated last month
- Benchmarking RL generalization in an interpretable way.☆156Updated 2 months ago
- Simple JAX Graphics Library.☆36Updated 6 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆236Updated last month
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆21Updated 4 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆143Updated last year
- cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.☆31Updated 9 months ago
- Episodic Control☆20Updated 2 years ago
- Code for magnetic mirror descent.☆16Updated last year
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆57Updated last year
- The Starcraft Multi-Agent challenge lite☆42Updated 8 months ago
- Example implemention of the Proximal Policy Optimization algorithm☆16Updated 9 months ago