Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.
☆16Aug 7, 2026Updated 3 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 6 months ago
- ☆19Apr 22, 2024Updated 2 years ago
- Official code release for ICLR26 "XQC: Well-conditioned Optimization Accelerates Deep Reinforcement Learning"☆35Jun 3, 2026Updated 2 months ago
- Implementations of Multi-Task and Meta-Learning baselines for the Metaworld benchmark☆39May 20, 2026Updated 3 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- 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☆11Aug 14, 2026Updated 2 weeks 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
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆32Oct 12, 2023Updated 2 years ago
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11Jul 12, 2026Updated last month
- 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.
- ☆20Oct 22, 2024Updated last year
- ☆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 4 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 4 months ago
- ☆19May 13, 2026Updated 3 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 3 years ago
- ☆20Jun 25, 2023Updated 3 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.
- This is the repo for DenseAttention and DANet - fast and conceptually simple modification of standard attention and Transformer☆20Apr 6, 2026Updated 4 months ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆74Aug 31, 2024Updated last year
- 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 5 years ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆27Updated this week
- 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.
- Comparison between GFlowNets & Maximum Entropy RL☆19Feb 19, 2024Updated 2 years ago
- Vintix: Action Model via In-Context Reinforcement Learning - - — ICML 2025☆51May 23, 2025Updated last year
- 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 11 months 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