Vybe Guide
Sign Up

Explore Projects

Discover 54 open source projects

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

Showing 21-40 of 54 projects

alpa-projects/alpa

Alpa is a distributed training and serving framework for large-scale neural networks with auto-parallelization.

3.2K
Archived
Python
LLM Frameworks
API Frameworks
JAX
#distributed-computing#high-performance-computing#auto-parallelization

NVIDIA/TransformerEngine

A high-performance Transformer library for accelerating AI models on NVIDIA GPUs, including low-precision support.

3.2K
Active
Python
LLM Frameworks
Inference
PyTorch
#deep-learning#gpu#cuda

grame-cncm/faust

Faust is a functional programming language for signal processing and sound synthesis.

3.0K
Active
C++
Backend Frameworks
CLI Tools
#audio#c#c-plus-plus

bethgelab/foolbox

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX.

2.9K
Stable
Python
Adversarial Attacks
Adversarial Examples
PyTorch
#adversarial-attacks#adversarial-examples#machine-learning

explosion/thinc

A refreshing functional take on deep learning, compatible with popular ML libraries like TensorFlow and PyTorch.

2.9K
Stable
Python
LLM Frameworks
CLI Tools
PyTorch
#machine-learning#deep-learning#functional-programming

danijar/dreamerv3

An open-source project that explores the use of world models for reinforcement learning in diverse domains like Minecraft.

2.9K
Stable
Python
Reinforcement Learning
World Models
Jax
#reinforcement-learning#world-models#minecraft

patrick-kidger/equinox

A Python library for building neural networks with JAX, providing an elegant and easy-to-use interface.

2.8K
Active
Python
LLM Frameworks
React
#neural-networks#jax#equinox

pyro-ppl/numpyro

A probabilistic programming library powered by NumPy and JAX for Bayesian inference and MCMC sampling.

2.6K
Active
Python
Inference
Caching
NumPy
#bayesian-inference#hmc#jax

google-deepmind/mctx

A JAX-based library for Monte Carlo tree search, a reinforcement learning technique useful for AI agents.

2.6K
Stable
Python
Reinforcement Learning
JAX
#jax#monte-carlo-tree-search#reinforcement-learning

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

pac4j/pac4j

A comprehensive security engine for Java applications, supporting various authentication and authorization frameworks.

2.5K
Active
Java
Authentication
Authorization
Java
#authentication#authorization#security

javaee-samples/javaee7-samples

A collection of Java EE 7 samples covering various technologies like CDI, JAX-RS, JSF, and more.

2.5K
Archived
Java
API Frameworks
Backend Frameworks
#java#javaee#javaee7

google/neural-tangents

Neural Tangents is a Python library for fast and easy infinite neural networks using JAX.

2.4K
Archived
Jupyter Notebook
LLM Frameworks
API Frameworks
JAX
#neural-networks#bayesian-inference#gaussian-processes

apple/axlearn

An extensible deep learning library that leverages JAX for high-performance ML research and production use cases.

2.3K
Active
Python
LLM Frameworks
CLI Tools
React
#deep-learning#jax#python

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX, a high-performance machine learning framework.

2.2K
Active
Python
ML Ops
Build Tools
#machine-learning#optimization#jax

EMI-Group/evox

Distributed GPU-accelerated framework for evolutionary computation and optimization algorithms.

2.0K
Active
Python
Evolutionary Algorithms
CLI Tools
PyTorch
#evolutionary-computation#optimization#gpu-acceleration

patrick-kidger/diffrax

Numerical differential equation solvers in JAX, with autodifferentiation and GPU support.

1.9K
Active
Python
ML Ops
Backend Frameworks
JAX
#differential-equations#dynamical-systems#neural-differential-equations

deepmodeling/deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

1.9K
Active
Python
ML Ops
Databases
Python
#computational-chemistry#molecular-dynamics#deep-learning

google-deepmind/penzai

A JAX-based research toolkit for building, editing, and visualizing neural networks for vibe coders.

1.9K
Experimental
Python
Fine-tuning
Interpretability
JAX
#neural-networks#jax#ai-research

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
13

Stay in the loop

Get weekly updates on trending AI coding tools and projects.