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☆53Mar 24, 2025Updated last year
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 4 years ago
- Automatic Functional Differentiation in JAX☆88Sep 18, 2025Updated 10 months ago
- 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
- 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.
- 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
- A collection of deep reinforcement learning algorithm implementations☆11Jan 9, 2020Updated 6 years ago
- 上海交通大学软件学院课程《应用系统体系架构》(SE3353)笔记☆11Feb 2, 2024Updated 2 years ago
- Music GAN - GANSynth preprocessing, ProGAN and DCGAN architecture☆11Jan 26, 2023Updated 3 years ago
- rabitq rust implementation☆11May 14, 2026Updated 3 months ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"