Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for jax haiku
haiku
x
jax
x
7 search results found
Get Started With Jax
⭐
487
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
Omd
⭐
23
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Haiku Mup
⭐
18
A port of muP to JAX/Haiku
Safejax
⭐
13
Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
Sb3 Jax Haiku
⭐
7
stable-baselines with JAX & Haiku
Jax Vqvae Gpt
⭐
6
Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.
Quantum Orthogonal Nn
⭐
5
JAX implementation of Classical and Quantum Algorithms for Orthogonal Neural Networks by (Kerenidis et al., 2021)
1-7 of 7 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.