This repository extends a basic MLM implementation to allow for efficiently conditioning on chained previous texts, in a tree; for e.g., a Reddit thread.
-
Updated
May 23, 2021 - Python
This repository extends a basic MLM implementation to allow for efficiently conditioning on chained previous texts, in a tree; for e.g., a Reddit thread.
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Searching for galaxy satellites by their impact on lens potential in strong gravitational lensing setup
Master thesis: JAT (Jraph Attention Networks), a deep learning architecture to predict the potential energy and forces of molecules. Adapts Graph Attention Networks (GATv2) within the Message Passing Neural Networks framework to computational chemistry in JAX
This is the official repository of the paper "RoCourseNet: Distributionally Robust Training of a Prediction Aware Recourse Model".
NXML is an eXtension for Machine Learning
implicit differentiation with jax
Using evolutionary strategies as an alternative for reinforcement learning in compiler code optimization control
A lightweight wrapper with standard boiler plate code to train flax models
Diffusion models for probabilistic programming
A Keras implementation of Blenderbot, a state-of-the-art open-domain chatbot model, supporting TensorFlow, PyTorch, and JAX backends seamlessly.
Differentiable and accelerated wavelet transform on the ball with JAX
Denoising diffusion operators in JAX/FLAX
Machine learning Calabi-Yau metrics with JAX
Diffusion Probabilistic Model in Jax
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."