an environment based on XLA for deep learning compiler optimization research.
☆24Mar 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☆52Mar 24, 2025Updated last year
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆30Jul 18, 2023Updated 2 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 4 years ago
- Improving Token-Based World Models with Parallel Observation Prediction (ICML 2024)☆14Feb 23, 2026Updated 4 months ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Dec 20, 2019Updated 6 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.
- Automatic Functional Differentiation in JAX☆88Sep 18, 2025Updated 9 months ago
- ☆13Jul 25, 2023Updated 2 years ago
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆51Apr 20, 2026Updated 2 months ago
- Official code of paper Understanding, Predicting and Better Resolving Q-Value Divergence in Offline-RL☆24Oct 30, 2023Updated 2 years ago
- An implementation of MuZero in JAX.☆58Nov 8, 2022Updated 3 years ago
- A unified robotic manipulation learning framework☆23Sep 4, 2025Updated 10 months ago
- A collection of deep reinforcement learning algorithm implementations☆11Jan 9, 2020Updated 6 years ago
- ☆32Jun 6, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ESM2 protein language models in JAX/Flax☆19Oct 10, 2022Updated 3 years ago
- rabitq rust implementation☆11May 14, 2026Updated last month
- ResearchDoom fork of the Chocolate Doom engine.☆16Oct 20, 2017Updated 8 years ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆126Nov 27, 2024Updated last year
- ☆22Mar 30, 2023Updated 3 years ago
- Code release for "Understanding Bias in Large-Scale Visual Datasets"☆24Dec 4, 2024Updated last year
- Music GAN - GANSynth preprocessing, ProGAN and DCGAN architecture☆11Jan 26, 2023Updated 3 years ago
- Code for the Change-Based Inference Paper (CBinfer)☆10Jan 7, 2019Updated 7 years ago
- 🔍 Code Search Tools & Experiments☆12Jun 4, 2026Updated last month
- 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.
- ☆13Jun 26, 2022Updated 4 years ago
- Reinforcement learning library in JAX.☆102Oct 22, 2023Updated 2 years ago
- Reinforcement learning with Rust☆14Jul 31, 2022Updated 3 years ago
- Regularization Matters in Policy Optimization☆21Nov 1, 2021Updated 4 years ago
- Trends of arxiv submissions counted from twitter/medium/reddit etc.☆39Dec 11, 2022Updated 3 years ago
- neon implementation of SegNet☆13Jan 3, 2023Updated 3 years ago
- Official PyTorch implementation of the paper: "Deep Audio Waveform Prior" (Interspeech 2022) https://arxiv.org/abs/2207.10441☆12Oct 25, 2022Updated 3 years ago
- 🎲 A Kotlin DSL for probabilistic programming.☆12Apr 8, 2022Updated 4 years ago
- Efficiently send large arrays across machines☆16Jul 24, 2024Updated last year
- 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.
- LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrations☆30May 21, 2025Updated last year
- ☆16Jun 1, 2023Updated 3 years ago
- ☆11Jun 4, 2024Updated 2 years ago
- Reinforcing General Reasoning without Verifiers☆102Jun 24, 2025Updated last year
- ☆12Jan 4, 2024Updated 2 years ago
- A lightweight, Pythonic, frontend for MLIR☆80Oct 21, 2023Updated 2 years ago
- [ICML 2021] The official PyTorch Implementations of Positive-Negative Momentum Optimizers.☆28Aug 30, 2022Updated 3 years ago