Skip to content
View hbq1's full-sized avatar

Block or report hbq1

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
Showing results

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️

Python 262 18 Updated Nov 16, 2024

The code and models for the paper: Switti: Designing Scale-Wise Transformers for Text-to-Image Synthesis

Jupyter Notebook 160 15 Updated Dec 29, 2024

Loopy belief propagation for factor graphs on discrete variables in JAX

Jupyter Notebook 142 11 Updated Oct 18, 2024

Optax is a gradient processing and optimization library for JAX.

Python 1,802 208 Updated Feb 16, 2025

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to…

Python 66 5 Updated Nov 21, 2024

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

Python 685 88 Updated Feb 4, 2025

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Python 263 25 Updated Feb 3, 2025
Python 111 12 Updated Feb 5, 2025

Monte Carlo tree search in JAX

Python 2,421 196 Updated Dec 11, 2024
Jupyter Notebook 8 2 Updated Dec 10, 2021

Label-Efficient Semantic Segmentation with Diffusion Models (ICLR'2022)

Python 681 61 Updated Apr 8, 2023

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.

Jupyter Notebook 801 47 Updated Aug 12, 2024
Jupyter Notebook 464 98 Updated Jan 28, 2025
Python 100 8 Updated Jun 25, 2024

Open source code for AlphaFold 2.

Python 13,213 2,334 Updated Jan 29, 2025

PIX is an image processing library in JAX, for JAX.

Python 401 28 Updated Feb 16, 2025

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

Python 460 30 Updated Jan 22, 2025

Code for the paper: Distilling the Knowledge from Conditional Normalizing Flows

Jupyter Notebook 9 Updated Aug 5, 2021

RL research on Android devices.

Python 1,049 81 Updated Feb 6, 2025

JMP is a Mixed Precision library for JAX.

Python 191 17 Updated Jan 30, 2025
Python 319 39 Updated Aug 22, 2023

Probabilistic reasoning and statistical analysis in TensorFlow

Jupyter Notebook 4,293 1,111 Updated Jan 29, 2025
Python 547 32 Updated Dec 14, 2024

JAX-based neural network library

Python 2,959 236 Updated Jan 29, 2025

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research

C++ 716 99 Updated Jan 30, 2025

A place to submit conda recipes before they become fully fledged conda-forge feedstocks

Python 727 5,116 Updated Feb 16, 2025

Machine learning algorithms for many-body quantum systems

Python 573 192 Updated Feb 10, 2025
Python 587 52 Updated Jul 11, 2024
Next
Showing results