JAX and Keras logo
GSoCArtificial IntelligenceLast accepted 2025

JAX and Keras

Python libraries for large-scale machine learning

4
Projects
1
Years accepted
4
In 2025

Projects by year

4

2025

4 acceptedShow
  • Benchmarking and Improving Large Language Model Serving Performance

    Accepted project · by 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

    Accepted project · by 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

    Accepted project · by 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

    Accepted project · by neel04

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