an environment based on XLA for deep learning compiler optimization research.
☆23Mar 7, 2023Updated 3 years ago
Alternatives and similar repositories for hloenv
Users that are interested in hloenv are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Exchange correlation functionals translated from libxc to jax☆53Mar 24, 2025Updated last year
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆31Jul 18, 2023Updated 3 years ago
- Improving Token-Based World Models with Parallel Observation Prediction (ICML 2024)☆14Feb 23, 2026Updated 7 months ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Dec 20, 2019Updated 6 years ago
- Automatic Functional Differentiation in JAX☆89Sep 18, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- ☆13Jul 25, 2023Updated 3 years ago
- ☆15Oct 13, 2025Updated 11 months ago
- Official code of paper Understanding, Predicting and Better Resolving Q-Value Divergence in Offline-RL☆24Oct 30, 2023Updated 2 years ago
- Imitation Learning via Differentiable Physics☆44Aug 6, 2022Updated 4 years ago
- An implementation of MuZero in JAX.☆58Nov 8, 2022Updated 3 years ago
- HrvvI's extension to PyTorch☆10Oct 12, 2022Updated 3 years ago
- A collection of deep reinforcement learning algorithm implementations☆11Jan 9, 2020Updated 6 years ago
- ☆32Jun 6, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- An efficient binary serialization format for numerical data.☆18Nov 3, 2025Updated 10 months ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆16May 10, 2024Updated 2 years ago
- ☆10Jul 14, 2018Updated 8 years ago
- ResearchDoom fork of the Chocolate Doom engine.☆16Aug 10, 2026Updated last month
- Play with agents and more.☆22Sep 18, 2023Updated 3 years ago
- A key-value storage system based on LSM Tree, using Skip List and Bloom Filter to accelerate.☆10Jun 10, 2021Updated 5 years ago
- ☆14Mar 26, 2020Updated 6 years ago
- Code for Unifying Gradient Estimators for Meta-Reinforcement Learning via Off-Policy Evaluation @ NeurIPS 2021☆13Nov 3, 2021Updated 4 years ago
- Large Language Agents Modulating Behaviour in Decentralized Autonomous Organizations☆24Jul 14, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A unified robotic manipulation learning framework☆24Sep 4, 2025Updated last year
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆52Apr 20, 2026Updated 5 months ago
- ☆93Nov 15, 2019Updated 6 years ago
- On demand communication☆34Apr 16, 2026Updated 5 months ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- Efficiently send large arrays across machines☆15Jul 24, 2024Updated 2 years ago
- ☆13Jun 26, 2022Updated 4 years ago
- 💬 Minimalistic repository to reproduce and serve CAMEL models.☆25Jun 26, 2023Updated 3 years ago
- 🎲 A Kotlin DSL for probabilistic programming.☆13Apr 8, 2022Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆16Apr 10, 2022Updated 4 years ago
- ☆12Nov 18, 2022Updated 3 years ago
- Zero Bubble Pipeline Parallelism☆466May 7, 2025Updated last year
- ☆10Jan 24, 2020Updated 6 years ago
- ☆21Updated this week
- A Haskell implementation of Peter Norvig's spelling corrector☆15Jun 5, 2015Updated 11 years ago
- A lightweight, Pythonic, frontend for MLIR☆80Oct 21, 2023Updated 2 years ago