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
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 4 years ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Dec 20, 2019Updated 6 years ago
- Automatic Functional Differentiation in JAX☆88Sep 18, 2025Updated 11 months 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.
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- Imitation Learning via Differentiable Physics☆44Aug 6, 2022Updated 4 years ago
- [EMNLP-2024] ⚓️ Sailor: Open Language Models for South-East Asia☆139Dec 21, 2024Updated last year
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Oct 25, 2021Updated 4 years ago
- An implementation of MuZero in JAX.☆58Nov 8, 2022Updated 3 years ago
- Vocabulary Parallelism☆26Mar 10, 2025Updated last year
- ESM2 protein language models in JAX/Flax☆19Oct 10, 2022Updated 3 years ago
- A collection of deep reinforcement learning algorithm implementations☆11Jan 9, 2020Updated 6 years ago
- ☆32Jun 6, 2024Updated 2 years 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.
- Music GAN - GANSynth preprocessing, ProGAN and DCGAN architecture☆11Jan 26, 2023Updated 3 years ago
- rabitq rust implementation☆11May 14, 2026Updated 3 months ago
- SJTU SE3331 CSE (a distributed file system with Raft and MapReduce)☆10Jan 14, 2024Updated 2 years ago
- ☆10Jul 14, 2018Updated 8 years ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆127Nov 27, 2024Updated last year
- ResearchDoom fork of the Chocolate Doom engine.☆16Aug 10, 2026Updated 3 weeks ago
- Code release for "Understanding Bias in Large-Scale Visual Datasets"☆25Dec 4, 2024Updated last year
- A key-value storage system based on LSM Tree, using Skip List and Bloom Filter to accelerate.☆10Jun 10, 2021Updated 5 years ago
- Play with agents and more.☆22Sep 18, 2023Updated 2 years 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.
- Code for Unifying Gradient Estimators for Meta-Reinforcement Learning via Off-Policy Evaluation @ NeurIPS 2021☆13Nov 3, 2021Updated 4 years ago
- A unified robotic manipulation learning framework☆24Sep 4, 2025Updated last year
- ☆11Jan 22, 2015Updated 11 years ago
- ☆93Nov 15, 2019Updated 6 years ago
- ☆13May 7, 2024Updated 2 years ago
- A "minimal" example of a Vulkan rainbow triangle in Odin with GLFW.☆13Jun 2, 2024Updated 2 years ago
- 🔍 Code Search Tools & Experiments☆13Updated this week
- Reinforcement learning with Rust☆14Jul 31, 2022Updated 4 years ago
- neon implementation of SegNet☆13Jan 3, 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.
- Reinforcement learning library in JAX.