Artificial Intelligence

JAX and Keras

Python libraries for large-scale machine learning

Technologies
AImachine learningPython
Topics
Jaxkerasmachine learningPYTHON LIBRARY
Total projects
4
Years participating
1
Most recent
2025

Participation history

1 program years
Projects by participation year 2025: 4 projects 4 2025

Historical projects

2025 4 projects

Official GSoC page ↗

Benchmarking and Improving Large Language Model Serving Performance

Contributor: Bob Chen

Large Language Model (LLM) serving frameworks like vLLM must deliver high throughput and low latency to justify the substantial cost of GPU and TPU...

Integrating VBLoRA into Keras

Contributor: Dhiraj BM

VB-LoRA (Vector Bank Low-Rank Adaptation) is a parameter-efficient fine-tuning technique that extends LoRA (Low-Rank Adaptation) by introducing a...

Implement open LLM models with JAX and Flax

Contributor: Megan Andrews

As interest in large language models continues to grow, many open-source implementations exist across various frameworks—but accessible, modular, and...

"Pomni" - JAX Code Assistant

Contributor: neel04

Current Frontier LLMs are really bad at assisting with writing JAX code due to the sparse resources spread across multiple different forums and...