flax

8 projets partagent ce topic GitHub

Les traits relient les membres réellement apparentés entre eux. La taille des points suit les étoiles.

🧬 Membres

Python code for "Probabilistic Machine learning" book by Kevin Murphy

uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam…

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning,…

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for…

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

JAX library for training sub-4B foundation models for edge

🔗 Familles voisines

jax 60 gym 39 colab 16 jupyter-notebooks 10 probabilistic-programming 13 pytorch-lightning 16 controllable-generation 8 generative-models 38

Mesuré à partir des topics GitHub communs aux deux projets, pondérés par leur rareté.