Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.
☆16Sep 1, 2026Updated 2 weeks ago
Alternatives and similar repositories for lox
Users that are interested in lox are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 9 months ago
- 🚀Simple and concise implementation of DQN on Atari Games in JAX🚀☆17Feb 4, 2026Updated 7 months ago
- ☆19Apr 22, 2024Updated 2 years ago
- ☆58Updated this week
- Official code release for ICLR26 "XQC: Well-conditioned Optimization Accelerates Deep Reinforcement Learning"☆35Jun 3, 2026Updated 3 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Implementations of Multi-Task and Meta-Learning baselines for the Metaworld benchmark☆39May 20, 2026Updated 3 months ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- An API for reinforcement learning agents.☆14Aug 26, 2026Updated 3 weeks ago
- GPU Based 3D Finite Volume Fluid code written in Julia☆11Aug 14, 2026Updated last month
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 3 years ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆16Feb 28, 2023Updated 3 years ago
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 years ago
- A Simple Car Simulator based on SFML Library and ROS☆13Apr 23, 2020Updated 6 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆33Oct 12, 2023Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11Jul 12, 2026Updated 2 months ago
- ☆20Oct 22, 2024Updated last year
- A haptic authoring tool☆12May 19, 2025Updated last year
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆59Apr 20, 2026Updated 5 months ago
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 3 years ago
- ☆20Jun 25, 2023Updated 3 years ago
- This is the repo for DenseAttention and DANet - fast and conceptually simple modification of standard attention and Transformer☆20Apr 6, 2026Updated 5 months ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆74Aug 31, 2024Updated 2 years ago
- Training non-differentiable networks via optimal transport.☆33Updated this week
- Simple visualized cetz editor☆12Jun 22, 2026Updated 2 months ago
- ☆66Jan 30, 2026Updated 7 months ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆43Aug 22, 2023Updated 3 years ago
- Some deep learning models written with mxnet and C++11.☆12Feb 6, 2018Updated 8 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 6 years ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆30Updated this week
- Comparison between GFlowNets & Maximum Entropy RL☆19Feb 19, 2024Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Vintix: Action Model via In-Context Reinforcement Learning - - — ICML 2025☆51May 23, 2025Updated last year
- Causal RL Environments Simulator☆38Aug 28, 2026Updated 3 weeks ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 3 years ago
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆26May 29, 2026Updated 3 months ago
- The official repository of Terra Nova, a comprehensive challenge environment for reinforcement learning research inspired by Civilization…☆26Nov 24, 2025Updated 9 months ago
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Gym env for Slay the Spire☆19Dec 31, 2024Updated last year