Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.
☆16Aug 7, 2026Updated this week
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 8 months ago
- 🚀Simple and concise implementation of DQN on Atari Games in JAX🚀☆17Feb 4, 2026Updated 6 months ago
- ☆19Apr 22, 2024Updated 2 years ago
- ☆54Jul 28, 2026Updated last week
- Official code release for ICLR26 "XQC: Well-conditioned Optimization Accelerates Deep Reinforcement Learning"☆32Jun 3, 2026Updated 2 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.
- 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.☆14Updated this week
- GPU Based 3D Finite Volume Fluid code written in Julia☆11Jan 16, 2024Updated 2 years ago
- 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
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11Jul 12, 2026Updated 3 weeks ago
- ☆20Oct 22, 2024Updated last year
- 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.
- ☆13May 27, 2025Updated last year
- A haptic authoring tool☆12May 19, 2025Updated last year
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆57Apr 20, 2026Updated 3 months ago
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated 2 years ago
- Platform to run interactive Reinforcement Learning agents in a Minecraft Server☆57Apr 21, 2026Updated 3 months ago
- ☆16May 13, 2026Updated 2 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 3 years ago
- ☆20Jun 25, 2023Updated 3 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 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.
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆74Aug 31, 2024Updated last year
- Simple visualized cetz editor☆12Jun 22, 2026Updated last month
- Training non-differentiable networks via optimal transport.☆30Aug 1, 2026Updated last week
- ☆65Jan 30, 2026Updated 6 months ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆43Aug 22, 2023Updated 2 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 5 years ago
- JAX library for neural operator and PDE foundation model training.☆26Updated this week
- Comparison between GFlowNets & Maximum Entropy RL☆19Feb 19, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Vintix: Action Model via In-Context Reinforcement Learning - - — ICML 2025☆51May 23, 2025Updated last year
- Causal RL Environments Simulator☆38Updated this week
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 3 years ago
- Repository for "Revisiting Non-Acyclic GFlowNets in Discrete Environments" (ICML 2025)☆14Oct 8, 2025Updated 10 months ago
- Mastering Massive Multi-Task Reinforcement Learning via Mixture-of-Expert Decision Transformer☆22Sep 18, 2025Updated 10 months ago
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆25May 29, 2026Updated 2 months ago
- The official repository of Terra Nova, a comprehensive challenge environment for reinforcement learning research inspired by Civilization…☆26Nov 24, 2025Updated 8 months ago