Vybe Guide
Sign Up

Explore Projects

Discover 5 open source projects

Active filters (1):
Search: flax×
Clear all

Showing 1-5 of 5 projects

google/flax

Flax is a flexible neural network library for JAX, designed for easy experimentation and research in machine learning.

7.1K
Active
Jupyter Notebook
LLM Frameworks
API Frameworks
JAX
#machine-learning#neural-networks#jax

probml/pyprobml

Python code for a book on probabilistic machine learning, with support for various ML frameworks.

7.0K
Stable
Jupyter Notebook
ML Ops
Books & Guides
PyTorch
#machine-learning#probabilistic-programming#jupyter-notebooks

FlaxEngine/FlaxEngine

Flax Engine is a powerful multi-platform 3D game engine that enables developers to create high-quality games and applications.

6.6K
Active
C#
Game Development
C#
#game-engine#multi-platform#3d-graphics

young-geng/EasyLM

EasyLM is a one-stop solution for pre-training, fine-tuning, evaluating, and serving large language models (LLMs) in JAX/Flax.

2.5K
Archived
Python
LLM Frameworks
LLM Wrappers & SDKs
JAX
#large-language-models#natural-language-processing#chatbot

yang-song/score_sde

Official code for a state-of-the-art score-based generative modeling approach using stochastic differential equations.

1.8K
Archived
Jupyter Notebook
Generative Models
API Frameworks
Flax
#diffusion-models#score-based-generative-modeling#stochastic-differential-equations

Stay in the loop

Get weekly updates on trending AI coding tools and projects.